Update app.py

app.py

@@ -517,8 +517,8 @@ def create_demo():
         with gr.Row():
             with gr.Column():
                 with gr.Row():
-                    input_fg = gr.Image(source='upload', type="numpy", label="Foreground Image", height=400)
-                    input_bg = gr.Image(source='upload', type="numpy", label="Background Image", height=400)
+                    input_fg = gr.Image(label="Foreground Image", height=400, type="numpy")
+                    input_bg = gr.Image(label="Background Image", height=400, type="numpy")
                 
                 prompt = gr.Textbox(label="Prompt", value="beautiful woman, cinematic lighting")
                 bg_source = gr.Radio(
@@ -553,7 +553,7 @@ def create_demo():
                     n_prompt = gr.Textbox(label="Negative Prompt", value='lowres, bad anatomy, bad hands, cropped, worst quality')
             
             with gr.Column():
-                result_gallery = gr.Gallery(label='Results', height=600, object_fit='contain')
+                result_gallery = gr.Gallery(label='Results', height=600, columns=2, rows=2)
         
         # Event handlers
         inputs = [input_fg, input_bg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, bg_source]