Update app.py

app.py

@@ -16,10 +16,10 @@ app = Flask(__name__)
 DETECTION_URL = "/v1/detect"
 
 
-@app.route('/',methods=["POST"])
+@app.route(DETECTION_URL,methods=["POST"])
 def predict():
     
-    model = torch.hub.load('ultralytics/yolov5', 'custom', path='best2.pt', force_reload=True)  # force_reload to recache
+    #model = torch.hub.load('ultralytics/yolov5', 'custom', path='best2.pt', force_reload=True)  # force_reload to recache
     
     '''if not request.method == "POST":
         return'''
@@ -31,11 +31,20 @@ def predict():
         img = Image.open(io.BytesIO(image_bytes))
 
         results = model(img, size=640)  # reduce size=320 for faster inference
-        return results.pandas().xyxy[0].to_json(orient="records")
-
+        results=results.pandas().xyxy[0].to_json(orient="records")
+        return f"{results}"
 
+'''
 if __name__ == "__main__":
 
-    subprocess.run(["export","FLASK_ENV","=","development"])
+    #subprocess.run(["export","FLASK_ENV","=","development"])
     app.run(host="0.0.0.0", port=7860)  # debug=True causes Restarting with stat
     #serve(app, host="0.0.0.0", port=7860)
+'''    
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Flask API exposing YOLOv5 model")
+    parser.add_argument("--port", default=7860, type=int, help="port number")
+    args = parser.parse_args()
+
+    model = torch.hub.load('ultralytics/yolov5', 'custom', path='best2.pt', force_reload=True)  # force_reload to recache
+    app.run(host="0.0.0.0", port=args.port,debug =True)  # debug=True causes Restarting with stat