Update app.py

app.py

@@ -24,6 +24,9 @@ import faiss
 import numpy as np
 from PIL import Image
 
+
+
+
 # Configure logging
 logging.basicConfig(
     level=logging.INFO,
@@ -200,9 +203,11 @@ class ModelManager:
             "image_processor": {
                 "model_id": "Salesforce/blip-image-captioning-base",
                 "processor": AutoProcessor,
-                "model": AutoModel,
+                "model": BlipForConditionalGeneration,  # Changed from AutoModel
                 "kwargs": {
-                    "cache_dir": str(cache_dir)
+                    "cache_dir": str(cache_dir),
+                    "device_map": "auto"  # Add device mapping
+                   }
                 }
             }
         }