Adjust gallery and rename yolos_button

app.py

@@ -93,11 +93,12 @@ if __name__ == "__main__":
             with gr.Row():
                 with gr.Column():
                     yolos_input = gr.Image(type="pil")
-                    yolos_button = gr.Button("Submit")
+                    yolos_submit = gr.Button("Submit")
+                yolos_output = gr.Gallery(label="Detected Objects", object_fit="scale-down", columns=3, scale=2,
+                                          show_share_button=False, allow_preview=False, selected_index=None)
 
-                yolos_output = gr.Gallery(label="Detected Objects", object_fit="scale-down", columns=3, scale=1,
-                                          show_share_button=False)
-            yolos_selected = gr.Number(label="Selected Image Index")
+            with gr.Row():
+                yolos_selected = gr.Number(label="Selected Image Index")
 
         with gr.Tab("Inference"):
             with gr.Row():
@@ -108,7 +109,7 @@ if __name__ == "__main__":
                 moon_output = gr.TextArea(label="Output")
 
         # --- YOLOS --- #
-        yolos_button.click(detect_objects, [yolos_input], yolos_output)
+        yolos_submit.click(detect_objects, [yolos_input], yolos_output)
         yolos_output.select(get_selected_index, None, yolos_selected)
 
         # --- Moondream --- #