Update app.py

app.py

@@ -155,20 +155,6 @@ def generate_config(base_model, models, layer_range, merge_method):
 from functools import partial
 from itertools import chain
 
-# Configure dropdown options
-BASE_MODELS = ["bert-base-uncased", "distilbert-base-uncased", ...] # Add other base models here
-MERGE_METHODS = ["linear", "slerp", ...] # Add other merge methods here
-LAYER_RANGE = range(32)
-
-# Create input objects
-input_base_model = gr.Dropdown(label='Base Model', choices=BASE_MODELS)
-input_models = gr.Dropdown(label='Models', choices=BASE_MODELS)
-input_layer_range = gr.Slider(minimum=0, maximum=32, step=1, label='Layer Range')
-input_merge_method = gr.Dropdown(label='Merge Method', choices=MERGE_METHODS)
-
-# Wrap generate_config in a partial function to fix the signature
-partial_generate_config = partial(generate_config, base_model=input_base_model, merge_method=input_merge_method)
-
 # Generate config block
 #gen_config_block = gr.Blocks()
 #with gen_config_block:
@@ -254,6 +240,21 @@ def merge(yaml_config: str, hf_token: str, repo_name: str) -> Iterable[List[Log]
 
 with gr.Blocks() as demo:
     gr.Markdown(MARKDOWN_DESCRIPTION)
+    
+    # Configure dropdown options
+    BASE_MODELS = ["bert-base-uncased", "distilbert-base-uncased", ...] # Add other base models here
+    MERGE_METHODS = ["linear", "slerp", ...] # Add other merge methods here
+    LAYER_RANGE = range(32)
+    
+    # Create input objects
+    input_base_model = gr.Dropdown(label='Base Model', choices=BASE_MODELS)
+    input_models = gr.Dropdown(label='Models', choices=BASE_MODELS)
+    input_layer_range = gr.Slider(minimum=0, maximum=32, step=1, label='Layer Range')
+    input_merge_method = gr.Dropdown(label='Merge Method', choices=MERGE_METHODS)
+    
+    # Wrap generate_config in a partial function to fix the signature
+    partial_generate_config = partial(generate_config, base_model=input_base_model, merge_method=input_merge_method)
+
     with gr.Row():
         generated_config = gr.Textbox(label='Generated Config', interactive=False)
         btn_generate_config = gr.Button('Generate Config', variant='secondary')