Update app.py

app.py

@@ -101,7 +101,7 @@ examples = [
         "What are the things I should be cautious about when I visit here?",
     ]
 ]
-def test(image, question):
+def chatbot(image, question):
     image = image_processor(images=image.convert('RGB'), return_tensors='pt')['pixel_values']
     image_features = visual_encoder.encode_images(image.unsqueeze(0))
     emb_mixer.set_image_embeddings(image_features)
@@ -116,15 +116,15 @@ with gr.Blocks(title=title) as demo:
         with gr.Column():
             image = gr.Image(type='pil', label="Image")
         with gr.Column():
-            prompt = gr.Textbox(lines=2, label="Prompt", 
+            prompt = gr.Textbox(lines=3, 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")
+                clear = gr.Button("Clear", variant="secondary") 
+        with gr.Column():
             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])
+    submit.click(chatbot, [image, prompt], [output])
     clear.click(lambda: None, [], [output])
     data.click(lambda x: x, [data], [image, prompt])