Update app.py

app.py

@@ -334,13 +334,14 @@ with gr.Blocks(css="custom.css") as demo:
                 text_button = gr.Button("Retrieve Images", elem_id="run_button")
             #i = gr.Image(elem_id="result-image", label = "Image upload or selected", height = 768 - 256 - 32)
             with gr.Row():
-                title = gr.Markdown(
-                        value="### Click on a Image in the gallery to select it",
-                        visible=True,
-                        elem_id="selected_model",
-                    )
-                choose_model = gr.Radio(choices=["0", "1", "2", "3"],
-                    value="0", label="Choose embedding layer", elem_id="layer_radio")
+                with gr.Column():
+                    title = gr.Markdown(
+                            value="### Click on a Image in the gallery to select it",
+                            visible=True,
+                            elem_id="selected_model",
+                        )
+                    choose_model = gr.Radio(choices=["0", "1", "2", "3"],
+                        value="0", label="Choose embedding layer", elem_id="layer_radio")
             with gr.Row():
                 g_outputs = gr.Gallery(label='Output gallery', elem_id="gallery",
                 columns=[5],object_fit="contain", height="auto")