Update app.py

app.py

@@ -2,11 +2,11 @@ import torch
 
 import gradio as gr
 
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import (PegasusForConditionalGeneration, PegasusTokenizer)
   
-tokenizer = AutoTokenizer.from_pretrained("aditi2222/paragus_models")
-
-model = AutoModelForSeq2SeqLM.from_pretrained("aditi2222/paragus_models")
+best_model_path = "/content/pegasus_paraphrase_model"
+model = PegasusForConditionalGeneration.from_pretrained(best_model_path)
+tokenizer = PegasusTokenizer.from_pretrained('google/pegasus-xsum')
 
 def tokenize_data(text):
     # Tokenize the review body