Spaces:
Sleeping
Sleeping
Commit
·
179f37d
1
Parent(s):
59d2287
Fix: yolo3 dependency
Browse files
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 =
|
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":
|