update

app.py

@@ -2,7 +2,7 @@ from utils import *
 import gradio as gr
 
 from sentence_transformers import SentenceTransformer
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModel
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 def download_model():
     # 下載並快取SentenceTransformer所需的模型和tokenizer
@@ -10,13 +10,13 @@ def download_model():
     
     # 下載並快取AutoTokenizer所需的模型
     biobart_model = "fuhsiao/BioBART-PMC-EXT-Section"
-    AutoModel.from_pretrained(biobart_model)
     AutoTokenizer.from_pretrained(biobart_model)
+    AutoModelForSeq2SeqLM.from_pretrained(biobart_model)
     
     # 下載並快取AutoModelForSeq2SeqLM所需的模型
     bart_model = "fuhsiao/BART-PMC-EXT-Section"
-    AutoModelForSeq2SeqLM.from_pretrained(bart_model)
     AutoTokenizer.from_pretrained(bart_model)
+    AutoModelForSeq2SeqLM.from_pretrained(bart_model)
     
     return True