Update app.py

app.py

@@ -7,7 +7,7 @@ import json
 
 # Model
 # model = torch.hub.load('ultralytics/yolov5', 'yolov5s')  # force_reload=True to update
-model = torch.hub.load("best.pt") 
+model = torch.hub.load("custom","best.pt") 
 
 def yolo(im, size=1024):
     g = (size / max(im.size))  # gain