BhumikaMak commited on
Commit
179f37d
·
1 Parent(s): 59d2287

Fix: yolo3 dependency

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def draw_detections(boxes, colors, names, img):
45
  # Load the appropriate YOLO model based on the version
46
  def load_yolo_model(version="yolov5"):
47
  if version == "yolov3":
48
- model = YOLO("yolov3n.pt")
49
  elif version == "yolov5":
50
  model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
51
  elif version == "yolov7":
 
45
  # Load the appropriate YOLO model based on the version
46
  def load_yolo_model(version="yolov5"):
47
  if version == "yolov3":
48
+ model = torch.hub.load('ultralytics/yolov3', 'yolov3', pretrained=True)
49
  elif version == "yolov5":
50
  model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
51
  elif version == "yolov7":