Update app.py
Browse files
app.py
CHANGED
|
@@ -64,6 +64,8 @@ else:
|
|
| 64 |
|
| 65 |
################ STATEMENT SUMMARIZATION1 - side bar - tokenizer #################
|
| 66 |
|
|
|
|
|
|
|
| 67 |
# Load the summarization model and tokenizer
|
| 68 |
MODEL_NAME = "facebook/bart-large-cnn" # A commonly used summarization model
|
| 69 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
|
|
|
| 64 |
|
| 65 |
################ STATEMENT SUMMARIZATION1 - side bar - tokenizer #################
|
| 66 |
|
| 67 |
+
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 68 |
+
|
| 69 |
# Load the summarization model and tokenizer
|
| 70 |
MODEL_NAME = "facebook/bart-large-cnn" # A commonly used summarization model
|
| 71 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|