reset

app/routers/image.py

@@ -18,14 +18,14 @@ async def handleImageRequest(
 ):
  try:
  img = imfrombytes(file, cv2.IMREAD_COLOR)
- except:
- raise Response(content="Failed to read image", status_code=400)
+ if raw:
+ bboxes, labels = inferenceImage(img, threshold, True)
+ return {"bboxes": bboxes.tolist(), "labels": labels.tolist()}
 
- if raw:
- bboxes, labels = inferenceImage(img, threshold, raw)
- return {"bboxes": bboxes.tolist(), "labels": labels.tolist()}
+ img = inferenceImage(img, threshold, False)
+ except:
+ return Response(content="Failed to read image", status_code=400)
 
- img = inferenceImage(img, threshold, raw)
  ret, jpeg = cv2.imencode(".jpg", img)
 
  if not ret: