Update app.py

app.py

@@ -33,17 +33,17 @@ id_model = Idefics3ForConditionalGeneration.from_pretrained("HuggingFaceM4/Idefi
 BAD_WORDS_IDS = id_processor.tokenizer(["<image>", "<fake_token_around_image>"], add_special_tokens=False).input_ids
 EOS_WORDS_IDS = [id_processor.tokenizer.eos_token_id]
 
-# # Load colpali model
-# model_name = "vidore/colpali-v1.2"
-# token = os.environ.get("HF_TOKEN")
-# model = ColPali.from_pretrained(
-#     "vidore/colpaligemma-3b-pt-448-base", torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
+# Load colpali model
+model_name = "vidore/colpali-v1.2"
+token = os.environ.get("HF_TOKEN")
+model = ColPali.from_pretrained(
+    "vidore/colpaligemma-3b-pt-448-base", torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
 
-# model.load_adapter(model_name)
-# model = model.eval()
-# processor = AutoProcessor.from_pretrained(model_name, token = token)
+model.load_adapter(model_name)
+model = model.eval()
+processor = AutoProcessor.from_pretrained(model_name, token = token)
 
-# mock_image = Image.new("RGB", (448, 448), (255, 255, 255))
+mock_image = Image.new("RGB", (448, 448), (255, 255, 255))
 
 @spaces.GPU
 def model_inference(