Spaces:

Dheenaoo7
/

getitem

Sleeping

dheena commited on May 9

Commit

58597bf

1 Parent(s): ea99a27

SYS-0000

Files changed (1) hide show

src/model.py CHANGED Viewed

@@ -10,7 +10,7 @@ import os
 device = "cpu"
-model, preprocess = clip.load("ViT-B/32", device=device, download_root=os.getcwd())
 def get_image_features(image: Image.Image) -> np.ndarray:
     """Extract CLIP features from an image."""

 device = "cpu"
+model, preprocess = clip.load("ViT-B/32", device=device, download_root="/tmp")
 def get_image_features(image: Image.Image) -> np.ndarray:
     """Extract CLIP features from an image."""