Spaces:

pranshh
/

ocr-assignment

Running

pranshh commited on Sep 30, 2024

Commit

03fd65f

verified ·

1 Parent(s): bf86837

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import os
 import tempfile
 # Load ColPali model
-RAG = RAGMultiModalModel.from_pretrained("vidore/colpali", device_map="cpu", torch_dtype=torch.float32)
 processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct")

 import tempfile
 # Load ColPali model
+RAG = RAGMultiModalModel.from_pretrained("vidore/colpali", torch_dtype=torch.float32)
 processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct")