madankn79 commited on
Commit
9388e0e
·
1 Parent(s): daec533
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -16,10 +16,9 @@ model_choices = {
16
  "T5 Small (t5-small)": "t5-small",
17
  "T5 Base (t5-base)": "t5-base",
18
  "Flan-T5 Base (google/flan-t5-base)": "google/flan-t5-base",
19
- "DistilBART XSum (mrm8488/distilbart-xsum-12-6)": "mrm8488/distilbart-xsum-12-6"
20
  }
21
 
22
-
23
  model_cache = {}
24
 
25
  def clean_text(input_text):
 
16
  "T5 Small (t5-small)": "t5-small",
17
  "T5 Base (t5-base)": "t5-base",
18
  "Flan-T5 Base (google/flan-t5-base)": "google/flan-t5-base",
19
+ "DistilBART XSum (sshleifer/distilbart-xsum-12-6)": "sshleifer/distilbart-xsum-12-6"
20
  }
21
 
 
22
  model_cache = {}
23
 
24
  def clean_text(input_text):