Spaces:
Running
Running
from fastapi import HTTPException,File,UploadFile | |
from .preprocess import preprocess_image | |
from .inferencer import classify_image | |
async def Classify_Image_router(file: UploadFile = File(...)): | |
try: | |
image_array = preprocess_image(file) | |
result = classify_image(image_array) | |
return result | |
except Exception as e: | |
raise HTTPException(status_code=400, detail=str(e)) | |