Zengyf-CVer commited on
Commit
e4ef1b8
1 Parent(s): 3e29748

add yolov5 _verbose

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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(model_path, model_name, force_reload=True, device=device)
 
 
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