Spaces:

Xindus
/

xindus_summarizer

Sleeping

madankn79 commited on May 1

Commit

2289b0c

1 Parent(s): 32a64aa

google

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,11 +22,16 @@ stop_words = set(stopwords.words("english")).union(extra_stopwords)
 # Model list
 model_choices = {
     "DistilBART CNN (sshleifer/distilbart-cnn-12-6)": "sshleifer/distilbart-cnn-12-6",
     "T5 Small (t5-small)": "t5-small",
-    "T5 Base (t5-base)": "t5-base",
-    "Pegasus XSum (google/pegasus-xsum)": "google/pegasus-xsum",
-    "BART CNN (facebook/bart-large-cnn)": "facebook/bart-large-cnn",
 }
 model_cache = {}

 # Model list
 model_choices = {
+    "T5 Base (t5-base)": "t5-base",
     "DistilBART CNN (sshleifer/distilbart-cnn-12-6)": "sshleifer/distilbart-cnn-12-6",
+    "DistilBART XSum (sshleifer/distilbart-xsum-12-6)": "sshleifer/distilbart-xsum-12-6",
     "T5 Small (t5-small)": "t5-small",
+    "Flan-T5 Base (google/flan-t5-base)": "google/flan-t5-base",
+    "BART Large CNN (facebook/bart-large-cnn)": "facebook/bart-large-cnn",
+    "PEGASUS XSum (google/pegasus-xsum)": "google/pegasus-xsum",
+    "BART Large XSum (facebook/bart-large-xsum)": "facebook/bart-large-xsum",
+    "DistilGPT-2 (distilgpt2)": "distilgpt2",
+    "BART Large SciTLDR (facebook/bart-large-scitldr)": "facebook/bart-large-scitldr"
 }
 model_cache = {}