Update app.py

app.py

@@ -490,13 +490,13 @@ with gr.Blocks(css=css) as demo:
                         label="Light Direction",
                         scale=1
                     )
-            # Changed from Gallery to Image
-            result = gr.ImageSlider(
-                label="Result", 
-                show_label=True, 
-                type="pil"
-            )
-            final_prompt = gr.Textbox(label="Processed prompt")
+            with gr.Column():
+                result = gr.ImageSlider(
+                    label="Result", 
+                    show_label=True, 
+                    type="pil"
+                )
+                final_prompt = gr.Textbox(label="Processed prompt")
             
         with gr.Row():
             prompt = gr.Text(