Styld
/

marqo-fashionSigLIP

Zero-Shot Image Classification

multimodal retrieval

Inference Endpoints

Model card Files Files and versions Community

Amitai Getzler commited on 29 days ago

Commit

93c7837

•

1 Parent(s): dc7652d

:dizzy: A

Files changed (1) hide show

handler.py +5 -1

handler.py CHANGED Viewed

@@ -10,8 +10,12 @@ from typing import Dict, Any
 class EndpointHandler:
     def __init__(self, path="hf-hub:Styld/marqo-fashionSigLIP"):
         self.model, self.preprocess_train, self.preprocess_val = (
-            open_clip.create_model_and_transforms("hf-hub:Styld/marqo-fashionSigLIP", device="cuda" if torch.cuda.is_available() else "cpu")
         )
         self.tokenizer = open_clip.get_tokenizer("hf-hub:Styld/marqo-fashionSigLIP")
     def classify_image(self, candidate_labels, image):

 class EndpointHandler:
     def __init__(self, path="hf-hub:Styld/marqo-fashionSigLIP"):
         self.model, self.preprocess_train, self.preprocess_val = (
+            open_clip.create_model_and_transforms("hf-hub:Styld/marqo-fashionSigLIP")
         )
+        if torch.cuda.is_available():
+            self.model = self.model.cuda()
         self.tokenizer = open_clip.get_tokenizer("hf-hub:Styld/marqo-fashionSigLIP")
     def classify_image(self, candidate_labels, image):