update app.py

app.py

@@ -170,7 +170,10 @@ with gr.Blocks() as demo:
 
     with gr.Tab("🦓 Grounding"):
         with gr.Row():
-            query_image = gr.Image(type="pil")
+            with gr.Column(scale=1):
+                query_image = gr.Image(type="pil")
+            with gr.Column(scale=1):
+                out_image = gr.Image(type="pil")
         with gr.Row():
             chat_input = gr.Textbox(lines=1, label="Chat Input")
         text_output = gr.Textbox(value="Output:", label="Model output")
@@ -178,10 +181,10 @@ with gr.Blocks() as demo:
         run_btn = gr.Button("Run model")
 
 
+        def on_click_fn(img, text): return generate(1, img, text)
 
-        def on_click_fn(img,text): return generate(0, img, text)
 
-        run_btn.click(on_click_fn, inputs=[query_image,chat_input], outputs=[text_output])
+        run_btn.click(on_click_fn, inputs=[query_image, chat_input], outputs=[text_output, out_image])
 
     with gr.Tab("🔢 Counting objects"):
         with gr.Row():