Update app.py

app.py

@@ -7,21 +7,24 @@ import json
 
 from sentence_transformers import SentenceTransformer
 
-sentence_transformer_model = SentenceTransformer("sentence-transformers/all-roberta-large-v1")
+
+
+
 
 
 nltk.download('punkt')
 with open('testbook.json') as f:
     test_book = json.load(f)
-
 tokenizer = AutoTokenizer.from_pretrained("UNIST-Eunchan/bart-dnc-booksum")
 
-
+@st.cache_resource
 def load_model(model_name):
+    nltk.download('punkt')
+    sentence_transformer_model = SentenceTransformer("sentence-transformers/all-roberta-large-v1")
     model = AutoModelForSeq2SeqLM.from_pretrained("UNIST-Eunchan/bart-dnc-booksum")
-    return model
+    return sentence_transformer_model, model
 
-model = load_model("UNIST-Eunchan/bart-dnc-booksum")
+sentence_transformer_model, model = load_model("UNIST-Eunchan/bart-dnc-booksum")
 
 def infer(input_ids, max_length, temperature, top_k, top_p):