Styld
/

marqo-fashionSigLIP

Zero-Shot Image Classification

multimodal retrieval

Inference Endpoints

Model card Files Files and versions Community

Amitai Getzler commited on Aug 22, 2024

Commit

02ca591

·

1 Parent(s): 76b5a9e

:heavy_plus_sign: Add

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="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'))
         )
         self.tokenizer = open_clip.get_tokenizer("hf-hub:Styld/marqo-fashionSigLIP")