Add AVIF support to CLIP server (#917)

* Add AVIF plugin as requirement to server * Add Pillow AVIF import
jina-ai · Jun 14, 2023 · c7e84a4 · c7e84a4
1 parent 6e418fe
commit c7e84a4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/server/clip_server/model/clip.py b/server/clip_server/model/clip.py
@@ -2,6 +2,7 @@
 
 import io
 
+import pillow_avif
 from PIL import Image
 from torchvision.transforms import Compose, Resize, CenterCrop, ToTensor, Normalize
 

diff --git a/server/setup.py b/server/setup.py
@@ -49,6 +49,7 @@
         'docarray<0.30.0',
         'prometheus-client',
         'open_clip_torch>=2.8.0',
+        'pillow-avif-plugin'
     ],
     extras_require={
         'onnx': [