Update app.py

app.py

@@ -411,7 +411,7 @@ def create_demo():
                 bg_source = gr.Radio(choices=[e.value for e in BGSource], value=BGSource.NONE.value, label="Preferenca e Ndriçimit", type='value')
                 aspect_ratio = gr.Radio(choices=["9:16", "1:1", "16:9"], value="1:1", label="Raporti i Aspektit")
                 relight_button = gr.Button(value="Rindriço")
-                result_gallery = gr.Gallery(label="Rezultatet", height=832, object_fit='contain', elem_classes="constrained-container")
+                result_image = gr.Image(label="Rezultati", type="numpy", height=480, width=480, elem_classes="constrained-container")
                 # Hidden components for other parameters and output_bg
                 image_width = gr.Slider(label="Gjerësia e Imazhit", minimum=256, maximum=1024, value=640, step=64, visible=False)
                 image_height = gr.Slider(label="Lartësia e Imazhit", minimum=256, maximum=1024, value=640, step=64, visible=False)
@@ -438,7 +438,7 @@ def create_demo():
             input_fg, prompt, image_width, image_height, num_samples, seed, steps,
             a_prompt, n_prompt, cfg, highres_scale, highres_denoise, lowres_denoise, bg_source
         ]
-        relight_button.click(fn=process_relight, inputs=ips, outputs=[output_bg, result_gallery])
+        relight_button.click(fn=process_relight, inputs=ips, outputs=[output_bg, result_image])
 
     return block