Update app.py

app.py

@@ -241,7 +241,7 @@ class mutil_model(nn.Module):
         return x
 
 
-pth_path = './res2net_pretrain_model_999.pt'
+pth_path = './res2net_model_6_new.pt'
 category_num = 9
 
 # "cuda" only when GPUs are available.
@@ -274,7 +274,7 @@ model.eval()
 #model.load_state_dict(model_state)  # 加载模型参数
 #model.eval()
 
-labels = ['中国风', '古典', '电子', '摇滚', '乡村', '说唱', '民谣', '动漫', '现代']
+labels = ['伤感', '开心', '励志', '宣泄', '平静', '感人']
 
 import requests
 import torch