examples added

app.py

@@ -6,7 +6,8 @@ import spaces
 from utils.annotate import annotate_with_boxes
 from utils.models import load_models, run_inference, CHECKPOINTS
 from utils.tasks import TASK_NAMES, TASKS, OBJECT_DETECTION_TASK_NAME, \
-    CAPTION_TASK_NAMES
+    CAPTION_TASK_NAMES, CAPTION_TASK_NAME, DETAILED_CAPTION_TASK_NAME, \
+    MORE_DETAILED_CAPTION_TASK_NAME
 
 MARKDOWN = """
 # Better Florence-2 Playground 🔥
@@ -26,12 +27,14 @@ MARKDOWN = """
 </div>
 """
 
-# OBJECT_DETECTION_EXAMPLES = [
-#     ["microsoft/Florence-2-large-ft", "Object Detection", "https://media.roboflow.com/notebooks/examples/dog-2.jpeg"]
-# ]
-# CAPTION_EXAMPLES = [
-#     ["microsoft/Florence-2-large-ft", "Caption", "https://media.roboflow.com/notebooks/examples/dog-2.jpeg"]
-# ]
+OBJECT_DETECTION_EXAMPLES = [
+    ["microsoft/Florence-2-large-ft", OBJECT_DETECTION_TASK_NAME, "https://media.roboflow.com/notebooks/examples/dog-2.jpeg"]
+]
+CAPTION_EXAMPLES = [
+    ["microsoft/Florence-2-large-ft", CAPTION_TASK_NAME, "https://media.roboflow.com/notebooks/examples/dog-2.jpeg"],
+    ["microsoft/Florence-2-large-ft", DETAILED_CAPTION_TASK_NAME, "https://media.roboflow.com/notebooks/examples/dog-2.jpeg"],
+    ["microsoft/Florence-2-large-ft", MORE_DETAILED_CAPTION_TASK_NAME, "https://media.roboflow.com/notebooks/examples/dog-2.jpeg"]
+]
 
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 MODELS, PROCESSORS = load_models(DEVICE)
@@ -54,6 +57,10 @@ def process(checkpoint_dropdown, task_dropdown, image_input):
         return response[task]
 
 
+image_output_component = None
+text_output_component = None
+
+
 with gr.Blocks() as demo:
     gr.Markdown(MARKDOWN)
     with gr.Row():
@@ -75,6 +82,7 @@ with gr.Blocks() as demo:
             @gr.render(inputs=task_dropdown_component)
             def show_output(text):
                 if text == OBJECT_DETECTION_TASK_NAME:
+                    global image_output_component
                     image_output_component = gr.Image(type='pil', label='Image Output')
                     submit_button_component.click(
                         fn=process,
@@ -86,6 +94,7 @@ with gr.Blocks() as demo:
                         outputs=image_output_component
                     )
                 elif text in CAPTION_TASK_NAMES:
+                    global text_output_component
                     text_output_component = gr.Textbox(label='Caption Output')
                     submit_button_component.click(
                         fn=process,
@@ -97,29 +106,31 @@ with gr.Blocks() as demo:
                         outputs=text_output_component
                     )
 
-    # @gr.render(inputs=task_dropdown_component)
-    # def show_examples(text):
-    #     if text == "Object Detection":
-    #         gr.Examples(
-    #             fn=process,
-    #             examples=OBJECT_DETECTION_EXAMPLES,
-    #             inputs=[
-    #                 checkpoint_dropdown_component,
-    #                 task_dropdown_component,
-    #                 image_input_component
-    #             ],
-    #             outputs=image_output_component
-    #         )
-    #     elif text == "Caption":
-    #         gr.Examples(
-    #             fn=process,
-    #             examples=CAPTION_EXAMPLES,
-    #             inputs=[
-    #                 checkpoint_dropdown_component,
-    #                 task_dropdown_component,
-    #                 image_input_component
-    #             ],
-    #             outputs=text_output_component
-    #         )
+    @gr.render(inputs=task_dropdown_component)
+    def show_examples(text):
+        if text == OBJECT_DETECTION_TASK_NAME:
+            global image_output_component
+            gr.Examples(
+                fn=process,
+                examples=OBJECT_DETECTION_EXAMPLES,
+                inputs=[
+                    checkpoint_dropdown_component,
+                    task_dropdown_component,
+                    image_input_component
+                ],
+                outputs=image_output_component
+            )
+        elif text in CAPTION_TASK_NAMES:
+            global text_output_component
+            gr.Examples(
+                fn=process,
+                examples=CAPTION_EXAMPLES,
+                inputs=[
+                    checkpoint_dropdown_component,
+                    task_dropdown_component,
+                    image_input_component
+                ],
+                outputs=text_output_component
+            )
 
 demo.launch(debug=False, show_error=True, max_threads=1)