Update app.py

app.py

@@ -9,9 +9,9 @@ import shutil
 # Clear the torch hub cache to prevent any previous corrupted downloads
 shutil.rmtree(torch.hub._get_torch_home(), ignore_errors=True)
 
-# Load the YOLOv5 model (ensure the path is correct)
+# Load the YOLOv5 model from the uploaded best.pt file
 model = torch.hub.load(
-    'ultralytics/yolov5', 'custom', path='yolov5/runs/train/exp/weights/best.pt', trust_repo=True
+    'ultralytics/yolov5', 'custom', path='best.pt', trust_repo=True
 )
 
 # Global variables to store images and labels for carousel functionality