zhouxiaoxi commited on
Commit
893a6eb
·
verified ·
1 Parent(s): bf33743

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ class Model(nn.Module):
31
  # 加载预训练模型
32
  bert_model = BertModel.from_pretrained("ckiplab/bert-base-chinese").to(device)
33
  model = Model(bert_model).to(device)
34
- model.load_state_dict(torch.load("/params/1bert.pt"))
35
  # 切换到eval模式
36
  model.eval()
37
 
 
31
  # 加载预训练模型
32
  bert_model = BertModel.from_pretrained("ckiplab/bert-base-chinese").to(device)
33
  model = Model(bert_model).to(device)
34
+ model.load_state_dict(torch.load("./params/1bert.pt"))
35
  # 切换到eval模式
36
  model.eval()
37