Update app.py

app.py

@@ -16,10 +16,10 @@ from tqdm import tqdm
 from transformers import AutoProcessor
 
 # Load model
-model_name = "vidore/colpali"
+model_name = "vidore/colpali-v1.2"
 token = os.environ.get("HF_TOKEN")
 model = ColPali.from_pretrained(
-    "google/paligemma-3b-mix-448", torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
+    "vidore/colpaligemma-3b-pt-448-base", torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
 
 model.load_adapter(model_name)
 model = model.eval()