cbir-image-similarity

Runtime error

Samuel Schmidt commited on Mar 7, 2023

Commit

564548c

1 Parent(s): cfa8379

Fix: Wrong url for clip model

Files changed (1) hide show

src/CLIP.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers import AutoProcessor, CLIPModel
 class CLIPImageEncoder:
     def __init__(self, device="cpu"):
         self.device = device
-        self.model = CLIPModel.from_pretrained("openai/clip-vit-large-patch32")
         self.processor = AutoProcessor.from_pretrained("openai/clip-vit-base-patch32")
     def encode_image(self, image_pil):

 class CLIPImageEncoder:
     def __init__(self, device="cpu"):
         self.device = device
+        self.model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
         self.processor = AutoProcessor.from_pretrained("openai/clip-vit-base-patch32")
     def encode_image(self, image_pil):