Update app.py

app.py

@@ -341,8 +341,8 @@ with gr.Blocks(css="custom.css") as demo:
                     )
                 choose_model = gr.Radio(choices=["0", "1", "2", "3"],
                     value="0", label="Choose embedding layer", elem_id="layer_radio")
-                g_outputs = gr.Gallery(label='Output gallery', elem_id="gallery",).style(grid=5,height=768 + 64 + 32,
-                    allow_preview=False, label = "retrieve Images")
+                g_outputs = gr.Gallery(label='Output gallery', elem_id="gallery",
+                columns=[5],object_fit="contain", height="auto")
             with gr.Row():
                 with gr.Tab(label = "Download"):
                     zip_button = gr.Button("Zip Images to Download", elem_id="zip_button")