Update app.py

app.py

@@ -388,7 +388,8 @@ def create_demo():
             prompt = gr.Textbox(label="Përshkrimi", placeholder="Shkruani përshkrimin këtu")
             bg_source = gr.Radio(choices=[e.value for e in BGSource], value=BGSource.NONE.value, label="Preferenca e Ndriçimit", type='value')
             relight_button = gr.Button(value="Rindriço")
-            # Hidden components for other parameters
+            result_gallery = gr.Gallery(label="Rezultatet", height=832, object_fit='contain')  # Visible output
+            # 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)
             num_samples = gr.Slider(label="Numri i Imazheve", minimum=1, maximum=12, value=1, step=1, visible=False)
@@ -400,7 +401,6 @@ def create_demo():
             highres_scale = gr.Slider(label="Shkalla e Rezolutës së Lartë", minimum=1.0, maximum=3.0, value=2, step=0.01, visible=False)
             highres_denoise = gr.Slider(label="Denoise i Rezolutës së Lartë", minimum=0.1, maximum=1.0, value=0.5, step=0.01, visible=False)
             lowres_denoise = gr.Slider(label="Denoise i Rezolutës së Ulët", minimum=0.1, maximum=1.0, value=0.9, step=0.01, visible=False)
-            result_gallery = gr.Gallery(label='Rezultatet', visible=False)
             output_bg = gr.Image(type="numpy", label="Parapërpunimi i Planit të Parë", visible=False)
 
         # Bind the relight button