Update app.py

app.py

@@ -20,7 +20,8 @@ from roboflow import Roboflow
 
 # Load your model
 #model = torch.hub.load('ultralytics/yolov8', 'custom', path='best.pt', trust_repo=True)
-model = YOLO('best.pt')
+#model = YOLO('best.pt')
+model = ultralytics.YOLO('yolov8m').load('best.pt')
 model.eval()
 
 # Define your classes