Update app.py
Browse files
app.py
CHANGED
@@ -213,7 +213,7 @@ def res2net152_v1b_26w_4s(pretrained=False, **kwargs):
|
|
213 |
model.load_state_dict(model_zoo.load_url(model_urls['res2net152_v1b_26w_4s']))
|
214 |
return model
|
215 |
|
216 |
-
pth_path = './
|
217 |
category_num = 7
|
218 |
|
219 |
|
|
|
213 |
model.load_state_dict(model_zoo.load_url(model_urls['res2net152_v1b_26w_4s']))
|
214 |
return model
|
215 |
|
216 |
+
pth_path = './res2net_pretrain_model_7_1.pt'
|
217 |
category_num = 7
|
218 |
|
219 |
|