case when no person

Dockerfile

@@ -11,4 +11,6 @@ RUN wget https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8l-
 
 COPY . .
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+
+# curl -X POST "https://unfinity-yolov8-pose.hf.space/predict-image" -F "[email protected]"

main.py

@@ -34,6 +34,9 @@ async def predict_image(file: UploadFile = File(...)):
     result = model_pose(input_image)[0]
     keypoints = utils.get_keypoints(result)
 
+    if not keypoints:
+        return JSONResponse(content={"error": "No person detected"})
+
     # draw keypoints
     output_image = draw_keypoints(input_image, keypoints).convert("RGB")
 
@@ -56,8 +59,3 @@ async def predict_image(file: UploadFile = File(...)):
     }
 
     return JSONResponse(content=json_data)
-
-
-# if __name__ == "__main__":
-#     import uvicorn
-#     uvicorn.run(app, host="0.0.0.0", port=7860)