Spaces:
Runtime error
Runtime error
Zengyf-CVer
commited on
Commit
•
e4ef1b8
1
Parent(s):
3e29748
add yolov5 _verbose
Browse files
app.py
CHANGED
@@ -86,7 +86,9 @@ def parse_args(known=False):
|
|
86 |
def model_loading(model_name, device):
|
87 |
|
88 |
# 加载本地模型
|
89 |
-
model = torch.hub.load(
|
|
|
|
|
90 |
|
91 |
return model
|
92 |
|
|
|
86 |
def model_loading(model_name, device):
|
87 |
|
88 |
# 加载本地模型
|
89 |
+
model = torch.hub.load(
|
90 |
+
model_path, model_name, force_reload=True, device=device, _verbose=False
|
91 |
+
)
|
92 |
|
93 |
return model
|
94 |
|