springml111 commited on
Commit
7cc4db4
1 Parent(s): 723026a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ from transformers import (PegasusForConditionalGeneration, PegasusTokenizer)
3
  import gradio as gr
4
 
5
  best_model_path = "springml111/Pegasus_Paraphrase_model"
6
- model = T5ForConditionalGeneration.from_pretrained(best_model_path)
7
- tokenizer = T5Tokenizer.from_pretrained("springml111/Pegasus_Paraphrase_model")
8
 
9
  def tokenize_data(text):
10
  # Tokenize the review body
 
3
  import gradio as gr
4
 
5
  best_model_path = "springml111/Pegasus_Paraphrase_model"
6
+ model = PegasusForConditionalGeneration.from_pretrained(best_model_path)
7
+ tokenizer = PegasusTokenizer.from_pretrained("springml111/Pegasus_Paraphrase_model")
8
 
9
  def tokenize_data(text):
10
  # Tokenize the review body