Update app.py

app.py

@@ -24,6 +24,23 @@ def main():
 
     model_options = {
         "distilbert-base-uncased-finetuned-sst-2-english": "DistilBERT (SST-2)",
+        "distilbert-base-uncased": "DistilBERT Uncased",
+        "bert-base-uncased": "BERT Uncased",
+        "bert-base-cased": "BERT Cased",
+        "bert-large-uncased": "BERT Large Uncased",
+        "bert-large-cased": "BERT Large Cased",
+        "roberta-base": "RoBERTa Base",
+        "roberta-large": "RoBERTa Large",
+        "albert-base-v2": "ALBERT Base v2",
+        "albert-large-v2": "ALBERT Large v2",
+        "google/electra-base-discriminator": "Electra Base Discriminator",
+        "google/electra-large-discriminator": "Electra Large Discriminator",
+        "xlnet-base-cased": "XLNet Base Cased",
+        "xlnet-large-cased": "XLNet Large Cased",
+        "gpt2": "GPT2",
+        "gpt2-medium": "GPT2 Medium",
+        "gpt2-large": "GPT2 Large",
+        "gpt2-xl": "GPT2 XL",
         # Add more models here if desired
     }