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
|