Update app.py

app.py

@@ -111,12 +111,22 @@ def test(image, question):
     for output in generate(input_text, image_ids):
         yield output
 
-demo = gr.ChatInterface(fn=test, 
-                        inputs=[gr.Image(type='pil'), "text"], 
-                        outputs="text", 
-                        examples=examples, 
-                        title=title, 
-                        description="VisualRWKV-v5.0")
+with gr.Blocks(title=title) as demo:
+    with gr.Row():
+        with gr.Column():
+            image = gr.Image(type='pil', label="Image")
+        with gr.Column():
+            prompt = gr.Textbox(lines=2, label="Prompt", 
+                value="Assistant: Please upload an image and ask a question.")
+        with gr.Column():
+            with gr.Row():
+                submit = gr.Button("Submit", variant="primary")
+                clear = gr.Button("Clear", variant="secondary")
+            output = gr.Textbox(label="Output", lines=5)
+    data = gr.Dataset(components=[image, prompt], samples=examples, label="Examples", headers=["Image", "Prompt"])
+    submit.click(test, [image, prompt], [output])
+    clear.click(lambda: None, [], [output])
+    data.click(lambda x: x, [data], [image, prompt])
 
-demo = demo.queue(concurrency_count=1, max_size=10)
-demo.launch(share=False, enable_queue=True)
+demo.queue(concurrency_count=1, max_size=10)
+demo.launch(share=False)