Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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模式
|