Update model_selector.py

model_selector.py

@@ -1,9 +1,11 @@
-from transformers import HUGGINGFACE_HUB_NAME, HUGGINGFACE_HUB_MODEL
+from transformers import AutoModelForSequenceClassification
 
 def get_model_list():
-    return [
+    # List of popular pre-trained models
+    models = [
         "bert-base-uncased",
         "distilbert-base-uncased",
         "roberta-base",
         "gpt2"
     ]
+    return models