Spaces:

geekyrakshit
/

medrag

Running

geekyrakshit commited on Oct 19, 2024

Commit

353a440

1 Parent(s): dc4625d

fix: catch byaldi import

Files changed (1) hide show

medrag_multi_modal/retrieval/colpali_retrieval.py CHANGED Viewed

@@ -2,7 +2,12 @@ import os
 from typing import Any, Optional
 import weave
-from byaldi import RAGMultiModalModel
 from PIL import Image
 import wandb

 from typing import Any, Optional
 import weave
+try:
+    from byaldi import RAGMultiModalModel
+except ImportError:
+    pass
 from PIL import Image
 import wandb