Update app.py

app.py

@@ -1,10 +1,10 @@
 import torch
 from transformers import PegasusForConditionalGeneration, PegasusTokenizer, AutoTokenizer, AutoModelForSeq2SeqLM
 
-model_name = 'geckos/pegasus-fined-tuned-on-paraphrase'
+model_name = 'prithivida/parrot_paraphraser_on_T5'
 torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
-tokenizer = PegasusTokenizer.from_pretrained(model_name)
-model = PegasusForConditionalGeneration.from_pretrained(model_name).to(torch_device)
+tokenizer = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name).to(torch_device)
 
 def get_response(input_text,num_return_sequences):
   batch = tokenizer.prepare_seq2seq_batch([input_text],truncation=True,padding='longest',max_length=60, return_tensors="pt").to(torch_device)