ui

app.py

@@ -37,6 +37,10 @@ class SimpleVLMInterface:
         top_p=0.95
     ):
         try:
+
+            if image is not None:
+                text_input = f"<image>{text_input}"
+
             # Prepare the input text
             messages = [{"role": "user", "content": text_input}]
             chat_input = self.processor.apply_chat_template(
@@ -52,6 +56,8 @@ class SimpleVLMInterface:
             else:
                 image_inputs = None
 
+
+
             # Prepare model inputs
             model_inputs = self.processor(
                 text=chat_input,