Update app.py
Browse files
app.py
CHANGED
@@ -8,6 +8,7 @@ def add_model_to_list(new_model):
|
|
8 |
if new_model and new_model not in model_list:
|
9 |
model_list.append(new_model)
|
10 |
return model_list
|
|
|
11 |
|
12 |
def create_config(model_name, num_labels, use_cache):
|
13 |
if model_name not in model_list:
|
@@ -18,7 +19,7 @@ def create_config(model_name, num_labels, use_cache):
|
|
18 |
with gr.Blocks() as demo:
|
19 |
gr.Markdown("## Config Class - Transformers")
|
20 |
with gr.Row():
|
21 |
-
model_dropdown = gr.Dropdown(label="Select a Model", choices=model_list, value=model_list[0])
|
22 |
new_model_input = gr.Textbox(label="Add a New Model", placeholder="Enter model name")
|
23 |
add_model_button = gr.Button("Add Model")
|
24 |
num_labels_input = gr.Number(label="Number of Labels", value=2)
|
|
|
8 |
if new_model and new_model not in model_list:
|
9 |
model_list.append(new_model)
|
10 |
return model_list
|
11 |
+
allow_custom_value=True
|
12 |
|
13 |
def create_config(model_name, num_labels, use_cache):
|
14 |
if model_name not in model_list:
|
|
|
19 |
with gr.Blocks() as demo:
|
20 |
gr.Markdown("## Config Class - Transformers")
|
21 |
with gr.Row():
|
22 |
+
model_dropdown = gr.Dropdown(label="Select a Model", choices=model_list, value=model_list[0], allow_custom_value=True)
|
23 |
new_model_input = gr.Textbox(label="Add a New Model", placeholder="Enter model name")
|
24 |
add_model_button = gr.Button("Add Model")
|
25 |
num_labels_input = gr.Number(label="Number of Labels", value=2)
|