update output

app.py

@@ -59,7 +59,7 @@ def run_example(task_prompt, image, text_input=None, model_id='dwb2023/florence2
         task=task_prompt,
         image_size=(image.width, image.height)
     )
-    return parsed_answer
+    return parsed_answer, generated_text
 
 def plot_bbox(image, data):
     fig, ax = plt.subplots()
@@ -119,9 +119,9 @@ def process_image(image, task_prompt, text_input=None, model_id='dwb2023/florenc
     image = Image.fromarray(image)  # Convert NumPy array to PIL Image
     if task_prompt == 'Object Detection':
         task_prompt = '<OD>'
-        results = run_example(task_prompt, image, model_id=model_id)
-        fig = plot_bbox(image, results['<OD>'])
-        return results, fig_to_pil(fig)
+        parsed_answer, generated_text = run_example(task_prompt, image, model_id=model_id)
+        fig = plot_bbox(image, parsed_answer['<OD>'])
+        return parsed_answer, generated_text, fig_to_pil(fig)
     else:
         return "", None  # Return empty string and None for unknown task prompts
 
@@ -148,6 +148,7 @@ with gr.Blocks(theme="sudeepshouche/minimalist") as demo:
                 submit_btn = gr.Button(value="Submit")
             with gr.Column():
                 output_text = gr.Textbox(label="Output Text")
+                parsed_text = gr.Textbox(label="Parsed Text")
                 output_img = gr.Image(label="Output Image")
 
         gr.Examples(
@@ -169,7 +170,7 @@ with gr.Blocks(theme="sudeepshouche/minimalist") as demo:
             label='Try examples'
         )
 
-        submit_btn.click(process_image, [input_img, task_prompt, model_selector], [output_text, output_img])
+        submit_btn.click(process_image, [input_img, task_prompt, model_selector], [output_text, parsed_text, output_img])
 
     gr.Markdown("## 🚀Other Cool Stuff:")
     gr.Markdown("- [Florence 2 Whitepaper](https://arxiv.org/pdf/2311.06242) - how I found out about the Roboflow 100 and the BCCD dataset.  Turns out this nugget was from the original [Florence whitepaper](https://arxiv.org/pdf/2111.11432) but useful all the same!")