Update app.py

app.py

@@ -28,8 +28,10 @@ class StopOnTokens(StoppingCriteria):
         return False
 
 @torch.no_grad()
-def response(message, history, image):
+def response(history, image):
+    gr.Info('Starting...')
     stop = StopOnTokens()
+    message = "please describe the image as an ecommerce product made out of wood"
 
     messages = [{"role": "system", "content": "You are a helpful assistant."}]
 
@@ -37,9 +39,6 @@ def response(message, history, image):
         messages.append({"role": "user", "content": user_msg})
         messages.append({"role": "assistant", "content": assistant_msg})
 
-    if len(messages) == 1:
-        message = f" <image>{message}"
-
     messages.append({"role": "user", "content": message})
 
     model_inputs = processor.tokenizer.apply_chat_template(
@@ -82,20 +81,15 @@ def response(message, history, image):
         history[-1][1] = partial_response
         yield history, gr.Button(visible=False), gr.Button(visible=True, interactive=True)
 
-def hello_world():
-    gr.Warning('This is a warning message.')
-    return "hello world"
-
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.HTML(title)
         image = gr.Image(type="pil")
-        submit = gr.Button(value="Upload", variant="primary")
         chat = gr.Chatbot(show_label=False)
-        message = gr.Textbox(interactive=True, show_label=False, container=False)
+        submit = gr.Button(value="Upload", variant="primary")
     response_handler = (
         response,
-        [message, chat, image],
+        [chat, image],
         [submit]
     )