Pichayada commited on
Commit
2d72dda
·
verified ·
1 Parent(s): 79ffe76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -4,13 +4,13 @@ from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer, AutoCon
4
 
5
  # List of summarization models
6
  model_names = [
7
- "google/bigbird-pegasus-large-arxiv",
8
- "facebook/bart-large-cnn",
9
- "google/t5-v1_1-large",
10
- "sshleifer/distilbart-cnn-12-6",
11
- "allenai/led-base-16384",
12
- "google/pegasus-xsum",
13
- "togethercomputer/LLaMA-2-7B-32K"
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