Update app.py
Browse files
app.py
CHANGED
|
@@ -4,13 +4,13 @@ from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer, AutoCon
|
|
| 4 |
|
| 5 |
# List of summarization models
|
| 6 |
model_names = [
|
| 7 |
-
"
|
| 8 |
-
"
|
| 9 |
-
"
|
| 10 |
-
"
|
| 11 |
-
"
|
| 12 |
-
"
|
| 13 |
-
"
|
| 14 |
]
|
| 15 |
|
| 16 |
# Placeholder for the summarizer pipeline, tokenizer, and maximum tokens
|
|
|
|
| 4 |
|
| 5 |
# List of summarization models
|
| 6 |
model_names = [
|
| 7 |
+
"distilbart-cnn-12-6",
|
| 8 |
+
"TinyLlama",
|
| 9 |
+
"phi-3-mini",
|
| 10 |
+
"Mistral-7B",
|
| 11 |
+
"BTLM-3B",
|
| 12 |
+
"flan-t5-small",
|
| 13 |
+
"MiniLM"
|
| 14 |
]
|
| 15 |
|
| 16 |
# Placeholder for the summarizer pipeline, tokenizer, and maximum tokens
|