Pichayada commited on
Commit
fc3b7a5
·
verified ·
1 Parent(s): 0b6e12b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -5,7 +5,9 @@ from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer, AutoCon
5
  # List of summarization models
6
  model_names = [
7
  "facebook/bart-large-cnn",
8
- "mrm8488/t5-base-finetuned-summarize-news",
 
 
9
  "google/flan-t5-small",
10
  "prithivMLmods/t5-Flan-Prompt-Enhance"
11
  ]
 
5
  # List of summarization models
6
  model_names = [
7
  "facebook/bart-large-cnn",
8
+ "mistralai/Mistral-7B-Instruct-v0.3",
9
+ "cnmoro/TeenyTinyLlama-460m-Summarizer-PTBR",
10
+ "mabrouk/amazon-review-summarizer-bart"
11
  "google/flan-t5-small",
12
  "prithivMLmods/t5-Flan-Prompt-Enhance"
13
  ]