Update pages/对话式文本检测工具.py
Browse files- pages/对话式文本检测工具.py +2 -2
pages/对话式文本检测工具.py
CHANGED
|
@@ -78,9 +78,9 @@ class Model(nn.Module):
|
|
| 78 |
|
| 79 |
def load_models_and_predict(text, device):
|
| 80 |
# 加载模型
|
| 81 |
-
MacBERT_base_CDialBias = torch.load('
|
| 82 |
MacBERT_base_CDialBias.to(device)
|
| 83 |
-
MacBERT_base_COLD = torch.load('
|
| 84 |
MacBERT_base_COLD.to(device)
|
| 85 |
|
| 86 |
# 获取密钥和ID
|
|
|
|
| 78 |
|
| 79 |
def load_models_and_predict(text, device):
|
| 80 |
# 加载模型
|
| 81 |
+
MacBERT_base_CDialBias = torch.load('models/MacBERT-base-CDialBias.pth')
|
| 82 |
MacBERT_base_CDialBias.to(device)
|
| 83 |
+
MacBERT_base_COLD = torch.load('models/MacBERT-base-CDialBias.pth')
|
| 84 |
MacBERT_base_COLD.to(device)
|
| 85 |
|
| 86 |
# 获取密钥和ID
|