Sandaruth commited on
Commit
e6ddaee
·
verified ·
1 Parent(s): f2af6e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ login(token=huggingface_token)
17
  models = {
18
  "T5_Full_FineTune_V0.1_40K": "ARSynopsis/T5_Full_FineTune_V0.1_40K",
19
  "T5_Full_FineTune_V0.1_80K": "ARSynopsis/T5_Full_FineTune_V0.1_80K",
20
- "BERT": "bert-base-uncased", # Note: BERT isn't designed for summarization; you can change this
21
  "LongT5": "google/long-t5-local-base",
22
  "Pegasus": "google/pegasus-xsum"
23
  }
 
17
  models = {
18
  "T5_Full_FineTune_V0.1_40K": "ARSynopsis/T5_Full_FineTune_V0.1_40K",
19
  "T5_Full_FineTune_V0.1_80K": "ARSynopsis/T5_Full_FineTune_V0.1_80K",
20
+ "BART_Base_Full_FineTune_V0.1_83K": "ARSynopsis/BART_Base_Full_FineTune_V0.1_83K",
21
  "LongT5": "google/long-t5-local-base",
22
  "Pegasus": "google/pegasus-xsum"
23
  }