madankn79 commited on
Commit
f91ed41
·
1 Parent(s): 631cb80
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -29,9 +29,7 @@ model_choices = {
29
  "Flan-T5 Base (google/flan-t5-base)": "google/flan-t5-base",
30
  "BART Large CNN (facebook/bart-large-cnn)": "facebook/bart-large-cnn",
31
  "PEGASUS XSum (google/pegasus-xsum)": "google/pegasus-xsum",
32
- "BART Large XSum (facebook/bart-large-xsum)": "facebook/bart-large-xsum",
33
- "DistilGPT-2 (distilgpt2)": "distilgpt2",
34
- "BART Large SciTLDR (facebook/bart-large-scitldr)": "facebook/bart-large-scitldr"
35
  }
36
 
37
  model_cache = {}
 
29
  "Flan-T5 Base (google/flan-t5-base)": "google/flan-t5-base",
30
  "BART Large CNN (facebook/bart-large-cnn)": "facebook/bart-large-cnn",
31
  "PEGASUS XSum (google/pegasus-xsum)": "google/pegasus-xsum",
32
+ "BART Large XSum (facebook/bart-large-xsum)": "facebook/bart-large-xsum"
 
 
33
  }
34
 
35
  model_cache = {}