Update app.py

app.py

@@ -1,10 +1,15 @@
 import torch
-from transformers import T5ForConditionalGeneration,T5Tokenizer, AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import streamlit as st
 st.title("Paraphrase")
 
-model = AutoModelForSeq2SeqLM.from_pretrained("eugenesiow/bart-paraphrase")
-tokenizer = AutoTokenizer.from_pretrained("eugenesiow/bart-paraphrase")  
+@st.cache(allow_output_mutation=True)
+def get_model():
+    tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
+    model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
+    return model, tokenizer
+    
+model, tokenizer = get_model()
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.to(device)