kuoyuan commited on
Commit
9f73203
1 Parent(s): 018cbbf

修改模型名称

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def collate(batch, tokenizer, useFirstDim=True):
19
 
20
  def launchGradioNLI(modelName: str):
21
  # 下载模型
22
- folder = huggingface_hub.snapshot_download('qiaokuoyuan/symptom-sick-2c')
23
 
24
  # 读取模型和tokenizer
25
  model = XLMRobertaForSequenceClassification.from_pretrained(folder)
 
19
 
20
  def launchGradioNLI(modelName: str):
21
  # 下载模型
22
+ folder = huggingface_hub.snapshot_download('qiaokuoyuan/model-symptom-to-sick-2c')
23
 
24
  # 读取模型和tokenizer
25
  model = XLMRobertaForSequenceClassification.from_pretrained(folder)