diff --git a/semantic_router/encoders/clip.py b/semantic_router/encoders/clip.py index c1b877792299e38a6764a7576757a94b68d60994..c7210707e44b65c261253253aa906c256435d856 100644 --- a/semantic_router/encoders/clip.py +++ b/semantic_router/encoders/clip.py @@ -66,7 +66,7 @@ class CLIPEncoder(BaseEncoder): "You can install it with: " "`pip install semantic-router[vision]`" ) - + try: from PIL import Image except ImportError: @@ -114,7 +114,7 @@ class CLIPEncoder(BaseEncoder): embeds = self._model.get_image_features(pixel_values=inputs) embeds = embeds.squeeze(0).cpu().detach().numpy() return embeds - + def _ensure_rgb(self, img: Any): rgbimg = self._Image.new("RGB", img.size) rgbimg.paste(img) diff --git a/semantic_router/encoders/vit.py b/semantic_router/encoders/vit.py index 9aba6257790103f9afc536bae971f2d66c58a243..801d29943c7416b25bc97a3cfb3a069fe13e6953 100644 --- a/semantic_router/encoders/vit.py +++ b/semantic_router/encoders/vit.py @@ -41,7 +41,7 @@ class VitEncoder(BaseEncoder): "You can install it with: " "`pip install semantic-router[vision]`" ) - + try: from PIL import Image except ImportError: