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 |
-
"distilbart-cnn-12-6",
|
| 8 |
-
"TinyLlama",
|
| 9 |
-
"
|
| 10 |
-
"Mistral-7B",
|
| 11 |
-
"
|
| 12 |
-
"flan-t5-small",
|
| 13 |
-
"MiniLM"
|
| 14 |
]
|
| 15 |
|
| 16 |
# Placeholder for the summarizer pipeline, tokenizer, and maximum tokens
|
|
|
|
| 4 |
|
| 5 |
# List of summarization models
|
| 6 |
model_names = [
|
| 7 |
+
"sshleifer/distilbart-cnn-12-6",
|
| 8 |
+
"TinyLlama/TinyLlama-1.1B",
|
| 9 |
+
"microsoft/Phi-3-mini-4k-instruct",
|
| 10 |
+
"mistralai/Mistral-7B-v0.1",
|
| 11 |
+
"cerebras/btlm-3b-8k-base",
|
| 12 |
+
"google/flan-t5-small",
|
| 13 |
+
"nreimers/MiniLM-L6-H384-uncased"
|
| 14 |
]
|
| 15 |
|
| 16 |
# Placeholder for the summarizer pipeline, tokenizer, and maximum tokens
|