zhouxiaoxi commited on
Commit
b2901f6
·
verified ·
1 Parent(s): abb51e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ names = ['负向', '正向']
9
  tokenizer = BertTokenizer.from_pretrained("bert-base-chinese")
10
 
11
  # 加载预训练模型
12
- bert_model = BertModel.from_pretrained("bert-base-chinese").to(device)
13
  model = Model(bert_model).to(device)
14
  model.load_state_dict(torch.load("/params/1bert.pt"))
15
  # 切换到eval模式
 
9
  tokenizer = BertTokenizer.from_pretrained("bert-base-chinese")
10
 
11
  # 加载预训练模型
12
+ bert_model = BertModel.from_pretrained("ckiplab/bert-base-chinese").to(device)
13
  model = Model(bert_model).to(device)
14
  model.load_state_dict(torch.load("/params/1bert.pt"))
15
  # 切换到eval模式