删除import

app.py

@@ -1,24 +1,8 @@
 import functools
-import os
-
 import huggingface_hub
-from transformers import XLMRobertaForSequenceClassification, XLMRobertaTokenizerFast, \
-    AutoModelForSequenceClassification, AutoTokenizer
-import random
-import paddlenlp.datasets
-from paddlenlp.data import Stack, Tuple, Pad
-from paddlenlp.transformers import ElectraForSequenceClassification, ElectraTokenizer, ElectraModel
+from transformers import XLMRobertaForSequenceClassification, XLMRobertaTokenizerFast
 import paddle
-import pandas as pd
-import re
-import tqdm
-import cachetools
-import typing
 import torch
-import torch.utils.data as Data
-import datetime
-from transformers import pipeline
-
 import gradio
 
 
@@ -33,7 +17,7 @@ def collate(batch, tokenizer, useFirstDim=True):
     return torch.LongTensor(input_ids), torch.LongTensor(labels)
 
 
-def launchGradioNLI():
+def launchGradioNLI(modelName: str):
     # 下载模型
     folder = huggingface_hub.snapshot_download('qiaokuoyuan/symptom-sick-2c')
 
@@ -71,4 +55,4 @@ def launchGradioNLI():
 
 
 if __name__ == '__main__':
-    launchGradioNLI()
+    launchGradioNLI(modelName=r'D:\modelFile\train\2c\nli\r210000')