rtzg

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(