Fix: yolo3 dependency

app.py

@@ -10,7 +10,9 @@ from pytorch_grad_cam import EigenCAM
 from pytorch_grad_cam.utils.image import show_cam_on_image, scale_cam_image
 from PIL import Image
 import gradio as gr
+from ultralytics import YOLO
 
+# Load a COCO-pretrained YOLOv3n model
 # Global Color Palette
 COLORS = np.random.uniform(0, 255, size=(80, 3))
 
@@ -43,7 +45,7 @@ def draw_detections(boxes, colors, names, img):
 # Load the appropriate YOLO model based on the version
 def load_yolo_model(version="yolov5"):
     if version == "yolov3":
-        model = torch.hub.load('ultralytics/yolov3', 'yolov3', pretrained=True)
+        model = YOLO("yolov3n.pt")
     elif version == "yolov5":
         model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
     elif version == "yolov7":

requirements.txt

@@ -5,4 +5,5 @@ numpy
 pillow
 opencv-python
 git+https://github.com/jacobgil/pytorch-grad-cam.git
-gradio
+gradio
+ultralytics