Update UI

Took time to figure this out in Gradio's playground instead of committing here every three minutes, lol

app.py

@@ -51,6 +51,11 @@ def answer_question(img, prompt):
 
 
 def detect_objects(img: Image.Image):
+    """
+    Submits an image to the YOLOS-Small-300 model for object detection.
+    :param img:
+    :return:
+    """
     inputs = yolos_processor(img, return_tensors="pt")
     outputs = yolos_model(**inputs)
 
@@ -115,22 +120,27 @@ if __name__ == "__main__":
                     with gr.Group():
                         yolos_gallery = gr.Gallery(label="Detected Objects", object_fit="scale-down", columns=3,
                                                    show_share_button=False, selected_index=None, allow_preview=False,
-                                                   type="pil", interactive=False, show_download_button=True)
+                                                   type="pil", interactive=False)
                         proceed_button = gr.Button("To Moondream", interactive=False)
 
             with gr.Tab("Inference", id='moondream'):
                 with gr.Row(equal_height=False):
-                    with gr.Group():
+                    with gr.Column():
                         moon_img = gr.Image(label="Image", type="pil", interactive=True, mirror_webcam=False)
-                        moon_prompt = gr.Textbox(label="Question", value="What is this food item?")
-                        moon_submit = gr.Button("Submit", interactive=False)
-                    moon_output = gr.TextArea(label="Output", interactive=False)
+                        with gr.Group():
+                            moon_prompt = gr.Textbox(label="Ask a question about the image:",
+                                                     value="What is this food item? Include any text on labels.")
+                            moon_submit = gr.Button("Submit", interactive=False, variant="primary")
+                    moon_output = gr.TextArea(label="Answer", interactive=False)
 
         # --- YOLOS --- #
         yolos_input.upload(lambda: gr.Button(interactive=True), None, yolos_submit)
         yolos_input.clear(lambda: gr.Button(interactive=False), None, yolos_submit)
 
-        yolos_submit.click(detect_objects, [yolos_input], yolos_gallery)
+        yolos_submit.click(detect_objects, yolos_input, yolos_gallery)
+        yolos_submit.click(lambda: (gr.Button(variant="secondary"), gr.Button(variant="primary")), None,
+                           [yolos_submit, proceed_button])
+
         yolos_gallery.select(gallery_selected, None, [selected_image, proceed_button])
         proceed_button.click(to_moondream, [yolos_gallery, selected_image], [tabs, moon_img, moon_submit])