Spaces:
Paused
Paused
Commit
•
8715807
1
Parent(s):
113ba48
change status code
Browse files
sources/Controllers/main.py
CHANGED
@@ -2,7 +2,7 @@ import os
|
|
2 |
|
3 |
import numpy as np
|
4 |
import yolov5
|
5 |
-
from fastapi import File, UploadFile
|
6 |
from fastapi.encoders import jsonable_encoder
|
7 |
from fastapi.responses import JSONResponse
|
8 |
from PIL import Image
|
@@ -80,11 +80,15 @@ async def upload(file: UploadFile = File(...)):
|
|
80 |
if 7 not in categories:
|
81 |
if len(categories) < 9:
|
82 |
error = "Missing fields! Detecting content failed!"
|
83 |
-
return JSONResponse(
|
|
|
|
|
84 |
elif 7 in categories:
|
85 |
if len(categories) < 10:
|
86 |
error = "Missing fields! Detecting content failed!"
|
87 |
-
return JSONResponse(
|
|
|
|
|
88 |
|
89 |
boxes = predictions[:, :4].tolist()
|
90 |
|
|
|
2 |
|
3 |
import numpy as np
|
4 |
import yolov5
|
5 |
+
from fastapi import File, UploadFile, status
|
6 |
from fastapi.encoders import jsonable_encoder
|
7 |
from fastapi.responses import JSONResponse
|
8 |
from PIL import Image
|
|
|
80 |
if 7 not in categories:
|
81 |
if len(categories) < 9:
|
82 |
error = "Missing fields! Detecting content failed!"
|
83 |
+
return JSONResponse(
|
84 |
+
status_code=status.HTTP_400_BAD_REQUEST, content={"message": error}
|
85 |
+
)
|
86 |
elif 7 in categories:
|
87 |
if len(categories) < 10:
|
88 |
error = "Missing fields! Detecting content failed!"
|
89 |
+
return JSONResponse(
|
90 |
+
status_code=status.HTTP_400_BAD_REQUEST, content={"message": error}
|
91 |
+
)
|
92 |
|
93 |
boxes = predictions[:, :4].tolist()
|
94 |
|