Zengyf-CVer commited on
Commit
400d72b
1 Parent(s): 8884040

model loading update

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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("ultralytics/yolov5", model_name, device=device)
 
 
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