Update pages/3_Abstractive Summary.py

pages/3_Abstractive Summary.py

@@ -16,10 +16,11 @@ def load_model():
     #model = PegasusForConditionalGeneration.from_pretrained(local_pegasus-large_tokenizer_model, max_position_embeddings=2000).to(torch_device)
     return model,tokenizer
 
+model,tokenizer = load_model()
 #run this the first time and use the local model for faster runtime
 tokenizer.save_pretrained("local_pegasus-large_tokenizer")
 model.save_pretrained("local_pegasus-large_tokenizer_model")
-model,tokenizer = load_model()
+
 
 st.header("Abstractive Summurization with PEGASUS-LARGE")
 st.text("Try inputting a prompt below!")