Update app.py

app.py

@@ -1,33 +1,32 @@
 import gradio as gr
 from transformers import AutoConfig
 
-# Predefined list of models
+# Initialize with some common models
 model_list = ["bert-base-uncased", "gpt2", "distilbert-base-uncased"]
 
-def update_model_list(new_model):
-    if new_model not in model_list:
+def add_model_to_list(new_model):
+    if new_model and new_model not in model_list:
         model_list.append(new_model)
     return model_list
 
 def create_config(model_name, num_labels, use_cache):
+    if model_name not in model_list:
+        model_list.append(model_name)
     config = AutoConfig.from_pretrained(model_name, num_labels=num_labels, use_cache=use_cache)
     return str(config)
 
 with gr.Blocks() as demo:
     gr.Markdown("## Config Class - Transformers")
     with gr.Row():
-        model_dropdown = gr.Dropdown(label="Select a Model", choices=model_list)
-        add_model_box = gr.Textbox(label="Add a New Model")
+        model_dropdown = gr.Dropdown(label="Select a Model", choices=model_list, value=model_list[0])
+        new_model_input = gr.Textbox(label="Add a New Model", placeholder="Enter model name")
         add_model_button = gr.Button("Add Model")
-    num_labels = gr.Number(label="Number of Labels", default=2)
-    use_cache = gr.Checkbox(label="Use Cache", default=True)
-    output = gr.Textbox(label="Config Output", readonly=True)
+    num_labels_input = gr.Number(label="Number of Labels", value=2)
+    use_cache_input = gr.Checkbox(label="Use Cache", value=True)
+    output_area = gr.Textbox(label="Config Output", readonly=True)
     submit_button = gr.Button("Create Config")
 
-    # Logic to add a new model to the dropdown
-    add_model_button.click(fn=update_model_list, inputs=add_model_box, outputs=model_dropdown)
-
-    # Logic to create a config
-    submit_button.click(fn=create_config, inputs=[model_dropdown, num_labels, use_cache], outputs=output)
+    add_model_button.click(fn=add_model_to_list, inputs=new_model_input, outputs=model_dropdown)
+    submit_button.click(fn=create_config, inputs=[model_dropdown, num_labels_input, use_cache_input], outputs=output_area)
 
 demo.launch()