Spaces:
Sleeping
Sleeping
Update Tokenizer
Browse files
app.py
CHANGED
|
@@ -47,7 +47,7 @@ summarize_button = col1.button("Summarize")
|
|
| 47 |
if summarize_button and user_input:
|
| 48 |
# Load pre-trained Pegasus model and tokenizer
|
| 49 |
model_name = "ale-dp/pegasus-finetuned-dialog-summarizer"
|
| 50 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
| 51 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
| 52 |
|
| 53 |
# Generate summary
|
|
|
|
| 47 |
if summarize_button and user_input:
|
| 48 |
# Load pre-trained Pegasus model and tokenizer
|
| 49 |
model_name = "ale-dp/pegasus-finetuned-dialog-summarizer"
|
| 50 |
+
tokenizer = AutoTokenizer.from_pretrained("google/pegasus-cnn_dailymail")
|
| 51 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
| 52 |
|
| 53 |
# Generate summary
|