Update app.py

app.py

@@ -38,6 +38,9 @@ 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(