file added

app.py

@@ -46,12 +46,7 @@ sentiment_analyzer = pipeline("sentiment-analysis")
 
 # Load pre-trained YOLOv5 model
 def load_yolov5_model():
-    model = torch.hub.load(
-        r'yolov5',
-        'custom',
-        path=r"models\best.pt",
-        source="local"
-    )
+    model = torch.hub.load(r"ultralytics/yolov5", 'custom', path=r'models\best.pt')
     model.eval()
     return model