hjp157688 commited on
Commit
8730ea4
·
verified ·
1 Parent(s): 7e7bb39

Update pages/对话式文本检测工具.py

Browse files
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('ROCAI\models\MacBERT-base-CDialBias.pth')
82
  MacBERT_base_CDialBias.to(device)
83
- MacBERT_base_COLD = torch.load('ROCAI\models\MacBERT-base-CDialBias.pth')
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