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

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