phuochungus commited on
Commit
8715807
1 Parent(s): 113ba48

change status code

Browse files
Files changed (1) hide show
  1. sources/Controllers/main.py +7 -3
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(status_code=402, content={"message": error})
 
 
84
  elif 7 in categories:
85
  if len(categories) < 10:
86
  error = "Missing fields! Detecting content failed!"
87
- return JSONResponse(status_code=402, content={"message": error})
 
 
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