fashxp
/

zero-shot-image-classification

Zero-Shot Image Classification

endpoints-template

Model card Files Files and versions

fashxp commited on Jan 25, 2024

Commit

0e7cf0c

·

1 Parent(s): e7656af

update params

Files changed (1) hide show

handler.py +1 -0

handler.py CHANGED Viewed

@@ -27,6 +27,7 @@ class EndpointHandler():
         parameters = data.pop("parameters", data)
         #parameters = inputs['parameters']
         candidate_labels = parameters['candidate_labels']
         # run prediction one image wit provided candiates
         prediction = self.pipeline(images=[image], candidate_labels=candidate_labels)

         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)