tech9-ai
/

colpali-v1.3-hf

Visual Document Retrieval

Model card Files Files and versions Community

adrish commited on 21 days ago

Commit

0d89ced

·

1 Parent(s): 16843db

update the code

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -13,7 +13,7 @@ class EndpointHandler:
         If no model path is provided, it defaults to 'vidore/colpali-v1.3-hf'.
         """
         if model_path is None:
-            model_path = "vidore/colpali-v1.3-hf"
         try:
             self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
             # Use the specialized ColPaliForRetrieval class.

         If no model path is provided, it defaults to 'vidore/colpali-v1.3-hf'.
         """
         if model_path is None:
+            model_path = os.path.dirname(os.path.realpath(__file__))
         try:
             self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
             # Use the specialized ColPaliForRetrieval class.