Update app.py

app.py

@@ -56,9 +56,18 @@ app = Flask(__name__)
 
 @app.route('/')
 def index():
-    return '<iframe frameBorder="0" height="100%" src="{}/?__dark-theme={}" width="100%"></iframe>'.format(
-    os.getenv('INACCEL_URL'),request.args.get('__dark-theme', 'false'))
+    '''return '<iframe frameBorder="0" height="100%" src="{}/?__dark-theme={}" width="100%"></iframe>'.format(
+    os.getenv('INACCEL_URL'),request.args.get('__dark-theme', 'false'))'''
+    if request.files.get("image"):
+        image_file = request.files["image"]
+        image_bytes = image_file.read()
+
+        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")
 
 
 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)