ClearML/clearml-serving
1
0
Fork 0
mirror of https://github.com/clearml/clearml-serving synced 2025-06-26 18:16:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[DEV] feature/bytes-payload | Add bytes as payload

Browse Source
This commit is contained in:
Aleksandar Ivanovski 2022-10-06 13:31:54 +02:00
parent 4b52103636
commit 09ed480bc2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clearml_serving/serving/main.py
View File

@ -5,7 +5,7 @@ import gzip
from fastapi import FastAPI, Request, Response, APIRouter, HTTPException from fastapi import FastAPI, Request, Response, APIRouter, HTTPException
from fastapi.routing import APIRoute from fastapi.routing import APIRoute
from typing import Optional, Dict, Any, Callable from typing import Optional, Dict, Any, Callable, Union
from clearml import Task from clearml import Task
from clearml_serving.version import __version__ from clearml_serving.version import __version__
@ -87,7 +87,7 @@ router = APIRouter(
@router.post("/{model_id}/{version}") @router.post("/{model_id}/{version}")
@router.post("/{model_id}/") @router.post("/{model_id}/")
@router.post("/{model_id}") @router.post("/{model_id}")
async def serve_model(model_id: str, version: Optional[str] = None, request: Dict[Any, Any] = None): async def serve_model(model_id: str, version: Optional[str] = None, request: Union[bytes, Dict[Any, Any]] = None):
try: try:
return_value = processor.process_request( return_value = processor.process_request(
base_url=model_id, base_url=model_id,