Spaces:
Runtime error
Runtime error
Commit
·
400d72b
1
Parent(s):
8884040
model loading update
Browse files
app.py
CHANGED
@@ -79,7 +79,9 @@ def model_loading(model_name, device):
|
|
79 |
|
80 |
# 加载本地模型
|
81 |
# model = torch.hub.load('ultralytics/yolov5', 'yolov5s', device=device)
|
82 |
-
model = torch.hub.load(
|
|
|
|
|
83 |
|
84 |
# model = torch.hub.load(
|
85 |
# local_model_path,
|
@@ -88,7 +90,7 @@ def model_loading(model_name, device):
|
|
88 |
# source="local",
|
89 |
# device=device,
|
90 |
# )
|
91 |
-
|
92 |
return model
|
93 |
|
94 |
|
|
|
79 |
|
80 |
# 加载本地模型
|
81 |
# model = torch.hub.load('ultralytics/yolov5', 'yolov5s', device=device)
|
82 |
+
model = torch.hub.load(
|
83 |
+
"ultralytics/yolov5", model_name, force_reload=True, device=device
|
84 |
+
)
|
85 |
|
86 |
# model = torch.hub.load(
|
87 |
# local_model_path,
|
|
|
90 |
# source="local",
|
91 |
# device=device,
|
92 |
# )
|
93 |
+
|
94 |
return model
|
95 |
|
96 |
|