Update app.py

app.py

@@ -1,7 +1,7 @@
 """
 Run a rest API exposing the yolov5s object detection model
 """
-import argparse
+
 import io
 import torch
 from flask import Flask, request
@@ -14,6 +14,9 @@ DETECTION_URL = "/v1/detect"
 
 @app.route('/')
 def predict():
+    
+    model = torch.hub.load('ultralytics/yolov5', 'custom', path='best2.pt', force_reload=True)  # force_reload to recache
+    
     if not request.method == "POST":
         return
 
@@ -28,5 +31,4 @@ def predict():
 
 
 if __name__ == "__main__":
-    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=7860)  # debug=True causes Restarting with stat