Update app.py

app.py

@@ -65,9 +65,9 @@ def clear_fn():
 with gr.Blocks(title="SD Models") as my_interface:
     with gr.Column():
         with gr.Row():
-            with gr.Column():
+            with gr.Column():  # Use Column instead of Row with scale
                 primary_prompt = gr.Textbox(label="Prompt", value="")
-            with gr.Column():
+            with gr.Column():  # Use Column instead of Row with scale
                 with gr.Row():
                     run = gr.Button("Run", variant="primary")
                     clear_btn = gr.Button("Clear")
@@ -76,7 +76,7 @@ with gr.Blocks(title="SD Models") as my_interface:
             model_idx = 1
             for model_path in models:
                 with gr.Column(scale=3, min_width=200):
-                    with gr.Box():
+                    with gr.Group():  # Changed from gr.Box() to gr.Group()
                         sd_outputs[model_idx] = gr.Image(label=model_path)
                     pass
                 model_idx += 1