Changed to using blocks.

app.py

@@ -119,7 +119,18 @@ prober = Prober(
     model_checkpoint= "cache/20211220_191132_refclef_32_512_resnet50_8_6_8_0.1_0.1_0.1_0_0.0001_0.0_12_4_90_1_0_0_0/best.ckpt"
 )
 
-demo = gr.Interface(fn=prober.probe, inputs=["number", "text"], outputs="image", live=True)
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            a = gr.Number(label="sample_id")
+            output = gr.Image(prober.preview_image(a))
+    with gr.Row():
+        re = gr.Textbox(label="referring expression")
+        greet_btn = gr.Button("Greet")
+        output = gr.Image()
+        greet_btn.click(fn=prober.probe, inputs=[a, name], outputs=output)
+
+# demo = gr.Interface(fn=prober.probe, inputs=["number", "text"], outputs="image", live=True)
 
 demo.queue(concurrency_count=10)
 demo.launch(debug=True)