Spaces:
Sleeping
Sleeping
Commit
•
7cc4db4
1
Parent(s):
723026a
Update app.py
Browse files
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 =
|
7 |
-
tokenizer =
|
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
|