Spaces:

zhouxiaoxi
/

text-sentiment-analysis

Runtime error

zhouxiaoxi commited on Aug 5, 2025

Commit

c1b60cd

verified ·

1 Parent(s): 0bd42e9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ class Model(nn.Module):
 # 加载预训练模型
 bert_model = BertModel.from_pretrained("ckiplab/bert-base-chinese").to(device)
 model = Model(bert_model).to(device)
-model.load_state_dict(torch.load("params/1bert.pt"))
 # 切换到eval模式
 model.eval()

 # 加载预训练模型
 bert_model = BertModel.from_pretrained("ckiplab/bert-base-chinese").to(device)
 model = Model(bert_model).to(device)
+model.load_state_dict(torch.load("params/1bert.pt", map_location=torch.device(device)))
 # 切换到eval模式
 model.eval()