Update app.py

app.py

@@ -28,10 +28,9 @@ class StopOnTokens(StoppingCriteria):
         return False
 
 @torch.no_grad()
-def response(history, image):
-    gr.Info('Starting...')
+def response(message, history, image):
+    gr.Info('Starting...' + message)
     stop = StopOnTokens()
-    message = 'please describe the image as an ecommerce product made out of wood'
 
     messages = [{"role": "system", "content": "You are a helpful assistant."}]
 
@@ -85,12 +84,13 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.HTML(title)
         image = gr.Image(type="pil")
+        message = gr.Textbox(interactive=True, show_label=False, container=False)
         chat = gr.Chatbot(show_label=False)
         submit = gr.Button(value="Upload", variant="primary")
         
     response_handler = (
         response,
-        [chat, image],
+        [message, chat, image],
         [submit]
     )