Update app.py

app.py

@@ -4,10 +4,14 @@ from PIL import Image
 import os
 import zipfile
 import io
+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)
 model = torch.hub.load(
-    'ultralytics/yolov5', 'custom', path='yolov5/runs/train/exp/weights/best.pt'
+    'ultralytics/yolov5', 'custom', path='yolov5/runs/train/exp/weights/best.pt', trust_repo=True
 )
 
 # Global variables to store images and labels for carousel functionality