Update app.py

app.py

@@ -156,10 +156,7 @@ from functools import partial
 from itertools import chain
 
 # Generate config block
-#gen_config_block = gr.Blocks()
-#with gen_config_block:
-#    generated_config = gr.Textbox(label='Generated Config', interactive=False)
-#    btn_generate_config = gr.Button('Generate Config', variant='secondary')
+
 #    btn_generate_config.click(fn=partial_generate_config, inputs=[input_base_model, input_models, input_layer_range], outputs=[generated_config])
 
 
@@ -241,21 +238,26 @@ 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():
+         # 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)
+        #gen_config_block = gr.Blocks()
+        #with gen_config_block:
+        #    generated_config = gr.Textbox(label='Generated Config', interactive=False)
+        #    btn_generate_config = gr.Button('Generate Config', variant='secondary')
         generated_config = gr.Textbox(label='Generated Config', interactive=False)
         btn_generate_config = gr.Button('Generate Config', variant='secondary')