Update app.py

app.py

@@ -25,22 +25,8 @@ 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"
+        "albert-base-v2": "ALBERT Base v2"
         # Add more models here if desired
     }