Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -241,7 +241,7 @@ class mutil_model(nn.Module):
|
|
241 |
return x
|
242 |
|
243 |
|
244 |
-
pth_path = './
|
245 |
category_num = 9
|
246 |
|
247 |
# "cuda" only when GPUs are available.
|
@@ -274,7 +274,7 @@ model.eval()
|
|
274 |
#model.load_state_dict(model_state) # 加载模型参数
|
275 |
#model.eval()
|
276 |
|
277 |
-
labels = ['
|
278 |
|
279 |
import requests
|
280 |
import torch
|
|
|
241 |
return x
|
242 |
|
243 |
|
244 |
+
pth_path = './res2net_model_6_new.pt'
|
245 |
category_num = 9
|
246 |
|
247 |
# "cuda" only when GPUs are available.
|
|
|
274 |
#model.load_state_dict(model_state) # 加载模型参数
|
275 |
#model.eval()
|
276 |
|
277 |
+
labels = ['伤感', '开心', '励志', '宣泄', '平静', '感人']
|
278 |
|
279 |
import requests
|
280 |
import torch
|