Update app.py

app.py

@@ -1,11 +1,10 @@
 import torch
-from transformers import PegasusForConditionalGeneration, PegasusTokenizer, AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 model_name = 'prithivida/parrot_paraphraser_on_T5'
 torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
-tokenizer = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name).to(torch_device)
-
+tokenizer = AutoTokenizer.from_pretrained("prithivida/parrot_paraphraser_on_T5")
+model = AutoModelForSeq2SeqLM.from_pretrained("prithivida/parrot_paraphraser_on_T5")
 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)
  translated = model.generate(**batch,max_length=60,num_beams=4, num_return_sequences=num_return_sequences, temperature=0.5)