zoya23 commited on
Commit
2f81c04
·
verified ·
1 Parent(s): 3333e39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
 
5
  # Set up model paths (you can later replace these with fine-tuned model folders)
6
  model_map = {
7
- "BART": "pavuluriashwitha/Bart_model",
8
  "T5": "t5-small",
9
  "PEGASUS": "google/pegasus-cnn_dailymail"
10
  }
 
4
 
5
  # Set up model paths (you can later replace these with fine-tuned model folders)
6
  model_map = {
7
+ "BART": "sshleifer/distilbart-cnn-12-6",
8
  "T5": "t5-small",
9
  "PEGASUS": "google/pegasus-cnn_dailymail"
10
  }