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
|