Update README.md

README.md

@@ -47,18 +47,18 @@ from tqdm import tqdm
 from transformers import AutoProcessor
 from PIL import Image
 
-from colpali_engine.models.paligemma_colbert_architecture import ColPali
+from colpali_engine.models.idefics_colbert_architecture import ColIdefics
 from colpali_engine.trainer.retrieval_evaluator import CustomEvaluator
-from colpali_engine.utils.colpali_processing_utils import process_images, process_queries
+from colpali_engine.utils.colidefics_processing_utils import process_images, process_queries
 from colpali_engine.utils.image_from_page_utils import load_from_dataset
 
 
 def main() -> None:
-    """Example script to run inference with ColPali"""
+    """Example script to run inference with ColIdefics"""
 
     # Load model
-    model_name = "vidore/colpali"
-    model = ColPali.from_pretrained("google/paligemma-3b-mix-448", torch_dtype=torch.bfloat16, device_map="cuda").eval()
+    model_name = "vidore/colidefics"
+    model = ColIdefics.from_pretrained("HuggingFaceM4/idefics2-8b", torch_dtype=torch.bfloat16, device_map="cuda").eval()
     model.load_adapter(model_name)
     processor = AutoProcessor.from_pretrained(model_name)