Update README.md

README.md

@@ -70,16 +70,15 @@ from PIL import Image
 
 from colpali_engine.models import ColPali, ColPaliProcessor
 
-model = cast(
-    ColPali,
-    ColPali.from_pretrained(
-        "vidore/colpali-v1.2",
-        torch_dtype=torch.bfloat16,
-        device_map="cuda:0",  # or "mps" if on Apple Silicon
-    ).eval(),
-)
-
-processor = cast(ColPaliProcessor, ColPaliProcessor.from_pretrained("vidore/colpali-v1.2"))
+model_name = "vidore/colpali-v1.2"
+
+model = ColPali.from_pretrained(
+    model_name,
+    torch_dtype=torch.bfloat16,
+    device_map="cuda:0",  # or "mps" if on Apple Silicon
+).eval()
+
+processor = ColPaliProcessor.from_pretrained(model_name)
 
 # Your inputs
 images = [