Styld
/

marqo-fashionSigLIP

Zero-Shot Image Classification

multimodal retrieval

Inference Endpoints

Model card Files Files and versions Community

Amitai Getzler commited on Aug 22

Commit

b76f035

•

1 Parent(s): 02ca591

:heavy_plus_sign: Dad

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -10,7 +10,7 @@ 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=torch.device('cuda'))
         )
         self.tokenizer = open_clip.get_tokenizer("hf-hub:Styld/marqo-fashionSigLIP")

 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=torch.device('cuda' if torch.cuda.is_available() else 'cpu'))
         )
         self.tokenizer = open_clip.get_tokenizer("hf-hub:Styld/marqo-fashionSigLIP")