Refactor input image and result gallery components for improved layout

app.py

@@ -407,7 +407,7 @@ with block:
         gr.Markdown(Intro)
     with gr.Row():
         with gr.Column():
-            input_image = gr.Image(source="upload", type="pil")
+            input_image = gr.Image(type="pil")
             user_prompt = gr.Textbox(label="User Prompt", value="")
             
             # Only show model selection if load_both_models is True
@@ -442,7 +442,7 @@ with block:
                 scale_factor = gr.Number(label="SR Scale", value=4)
             gr.Examples(examples=exaple_images, inputs=[input_image])
         with gr.Column():
-            result_gallery = gr.Gallery(label="Output", show_label=False, elem_id="gallery").style(grid=1)
+            result_gallery = gr.Gallery(label="Output", show_label=False, elem_id="gallery", columns=1)
             with gr.Row():
                 run_llava_button = gr.Button(value="Run LLAVA", label="Run LLAVA")
                 run_sr_button = gr.Button(value="Run DiT4SR", label="Run DiT4SR")