Update app.py

app.py

@@ -27,20 +27,21 @@ def process_image(image):
         if not isinstance(image, Image.Image):
             image = Image.fromarray(image)
         
-        inputs = processor(
-            images=image,
+        # Process image with proper settings
+        inputs = processor.process_images(  # Changed from processor()
+            images=[image],
             return_tensors="pt"
         ).to(model.device)
         
         logger.info("Generating embeddings")
         with torch.no_grad():
             outputs = model(**inputs)
-            embeddings = outputs.last_hidden_state.mean(dim=1).cpu().numpy()
+            embeddings = outputs.cpu().numpy()
             
         logger.info(f"Embeddings shape: {embeddings.shape}")
         return {
-            "embeddings": embeddings.tolist(),
-            "shape": embeddings.shape
+            "mixed_embedding_p1": embeddings[0, :1024].tolist(),
+            "mixed_embedding_p2": embeddings[0, 1024:].tolist(),
         }
     except Exception as e:
         logger.error(f"Error: {str(e)}", exc_info=True)