Update app.py

app.py

@@ -1,9 +1,10 @@
 import torch
-from transformers import T5ForConditionalGeneration,T5Tokenizer, AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import T5ForConditionalGeneration,T5Tokenizer, AutoTokenizer, AutoModelForSeq2SeqLM, PegasusTokenizer, PegasusForConditionalGeneration
 import streamlit as st
 
-tokenizer = AutoTokenizer.from_pretrained("Vamsi/T5_Paraphrase_Paws")  
-model = AutoModelForSeq2SeqLM.from_pretrained("Vamsi/T5_Paraphrase_Paws")
+model_name = 'tuner007/pegasus_paraphrase'
+tokenizer = PegasusTokenizer.from_pretrained(model_name)
+model = PegasusForConditionalGeneration.from_pretrained(model_name)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.to(device)
 
@@ -17,7 +18,7 @@ def translate_to_english(model, tokenizer, text):
     do_sample=True,
     max_length=256,
     top_k=120,
-    temperature=1.2,
+    temperature=1.5,
     top_p=0.98,
     early_stopping=True,
     num_return_sequences=10