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
|