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