update

app.py

@@ -1,26 +1,22 @@
 from utils import *
 import gradio as gr
 
-
-from transformers import PreTrainedModel, PreTrainedTokenizer, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModel
+from sentence_transformers import SentenceTransformer
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModel
 
 def download_model():
     # 下載並快取SentenceTransformer所需的模型和tokenizer
-    sentence_transformer_model = "sentence-transformers/all-MiniLM-L6-v2"
-    PreTrainedModel.from_pretrained(sentence_transformer_model)
-    PreTrainedTokenizer.from_pretrained(sentence_transformer_model)
+    SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
     
     # 下載並快取AutoTokenizer所需的模型
     biobart_model = "fuhsiao/BioBART-PMC-EXT-Section"
-    PreTrainedModel.from_pretrained(biobart_model)
-    AutoTokenizer.from_pretrained(biobart_model)
     AutoModel.from_pretrained(biobart_model)
+    AutoTokenizer.from_pretrained(biobart_model)
     
     # 下載並快取AutoModelForSeq2SeqLM所需的模型
     bart_model = "fuhsiao/BART-PMC-EXT-Section"
-    PreTrainedModel.from_pretrained(bart_model)
-    AutoTokenizer.from_pretrained(bart_model)
     AutoModelForSeq2SeqLM.from_pretrained(bart_model)
+    AutoTokenizer.from_pretrained(bart_model)
     
     return True
 
@@ -28,6 +24,7 @@ def download_model():
 
 
 
+
 def main(file, ext_threshold, article_type):
     
     if file is None or ext_threshold is None or article_type is None: