Update app.py

app.py

@@ -168,11 +168,20 @@ def extract_coordinates(text):
 
 def cpu_inference(conversation, model, tokenizer, processor):
     try:
+        prompt = processor.apply_chat_template(
+            conversation, 
+            tokenize=False,
+            add_generation_prompt=True
+        )
+
+        image = conversation[1]["content"][0]["image"]
         inputs = processor(
-            conversation,
+            text=[prompt],
+            images=[image],
             return_tensors="pt",
             padding=True,
-            truncation=True
+            truncation=True,
+            max_length=512
         )
 
         with torch.no_grad():