rename

tool.py

@@ -238,7 +238,7 @@ class VisualRAGTool(Tool):
 
         return pages
     
-    def _embed_images(self, pages: List[Page]) -> List[torch.Tensor]:
+    def compute_embeddings(self, pages: List[Page]) -> List[torch.Tensor]:
         """Embeds the images using the model."""
         """Example script to run inference with ColPali (ColQwen2)"""
         # run inference - docs
@@ -268,7 +268,7 @@ class VisualRAGTool(Tool):
         pgs = self.preprocess(files, contextualize=contextualize, api_key=api_key or self.api_key)
 
         # Embed the images
-        embds = self._embed_images(pgs)
+        embds = self.compute_embeddings(pgs)
 
         # Overwrite the database if necessary
         if overwrite_db: