Update app.py

app.py

@@ -170,11 +170,11 @@ input_merge_method = gr.Dropdown(label='Merge Method', choices=MERGE_METHODS)
 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:
-    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])
+#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])
 
 
 def merge(yaml_config: str, hf_token: str, repo_name: str) -> Iterable[List[Log]]:
@@ -255,7 +255,10 @@ def merge(yaml_config: str, hf_token: str, repo_name: str) -> Iterable[List[Log]
 with gr.Blocks() as demo:
     gr.Markdown(MARKDOWN_DESCRIPTION)
     with gr.Row():
-        gen_config_block
+        generated_config = gr.Textbox(label='Generated Config', interactive=False)
+        btn_generate_config = gr.Button('Generate Config', variant='secondary')
+        
+
     with gr.Row():
         
         filename = gr.Textbox(visible=False, label="filename")
@@ -284,7 +287,7 @@ with gr.Blocks() as demo:
         outputs=[config],
     )
     gr.Markdown(MARKDOWN_ARTICLE)
-
+    btn_generate_config.click(fn=partial_generate_config, inputs=[input_base_model, input_models, input_layer_range], outputs=[generated_config])
     button.click(fn=merge, inputs=[config, token, repo_name], outputs=[logs])