EMaghakyan
/

fashion-clip

Zero-Shot Image Classification

Model card Files Files and versions

EMaghakyan commited on Nov 8, 2023

Commit

fc3fbd2

·

1 Parent(s): 24b7e35

Fix init

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -6,7 +6,7 @@ import requests
 class EndpointHandler:
-    def __init__(self):
         self.model = CLIPModel.from_pretrained("patrickjohncyh/fashion-clip")
         self.processor = AutoProcessor.from_pretrained("patrickjohncyh/fashion-clip")
         self.tokenizer = AutoTokenizer.from_pretrained("openai/clip-vit-base-patch32")

 class EndpointHandler:
+    def __init__(self, path=""):
         self.model = CLIPModel.from_pretrained("patrickjohncyh/fashion-clip")
         self.processor = AutoProcessor.from_pretrained("patrickjohncyh/fashion-clip")
         self.tokenizer = AutoTokenizer.from_pretrained("openai/clip-vit-base-patch32")