Spaces:
Sleeping
Sleeping
changed model path
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ for i, url in enumerate(file_urls):
|
|
23 |
f"image_{i}.jpg"
|
24 |
)
|
25 |
|
26 |
-
model = torch.hub.load("ultralytics/yolov5", "custom", path="
|
27 |
force_reload=False)
|
28 |
|
29 |
path = [['image_0.jpg'], ['image_1.jpg']]
|
|
|
23 |
f"image_{i}.jpg"
|
24 |
)
|
25 |
|
26 |
+
model = torch.hub.load("ultralytics/yolov5", "custom", path="yolov5_0.65map_exp7_best.pt",
|
27 |
force_reload=False)
|
28 |
|
29 |
path = [['image_0.jpg'], ['image_1.jpg']]
|