Update app.py

app.py

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