fashxp
/

zero-shot-image-classification

Zero-Shot Image Classification

endpoints-template

Inference Endpoints

Model card Files Files and versions Community

fashxp commited on Jan 25

Commit

d793007

•

1 Parent(s): 0e7cf0c

update params

Files changed (1) hide show

handler.py +3 -1

handler.py CHANGED Viewed

@@ -27,8 +27,10 @@ class EndpointHandler():
         parameters = data.pop("parameters", data)
         #parameters = inputs['parameters']
         candidate_labels = parameters['candidate_labels']
         candidate_labels_array = list(map(str.strip, candidate_labels.split(',')))
         # run prediction one image wit provided candiates
-        prediction = self.pipeline(images=[image], candidate_labels=candidate_labels)
         return prediction[0]

         parameters = data.pop("parameters", data)
         #parameters = inputs['parameters']
         candidate_labels = parameters['candidate_labels']
+        print(candidate_labels)
         candidate_labels_array = list(map(str.strip, candidate_labels.split(',')))
         # run prediction one image wit provided candiates
+        prediction = self.pipeline(images=[image], candidate_labels=candidate_labels_array)
         return prediction[0]