Update app.py

app.py

@@ -6,9 +6,10 @@ import shutil
 
 # Load the model
 model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
+model2 = path='best.pt
 
 def detect_image(image):
-    results = model(image)
+    results = model2(image)
     return results.render()[0]