Thomasboosinger
/

owlvit-base-patch32

Zero-Shot Object Detection

Inference Endpoints

Model card Files Files and versions Community

Thomasboosinger commited on Feb 17

Commit

d602a6c

•

1 Parent(s): 9ba309e

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -22,5 +22,5 @@ class EndpointHandler():
         image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
         # run prediction one image wit provided candiates
-        detector = self.pipeline(images=[image], candidate_labels=inputs["candiates"])
         return detector

         image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
         # run prediction one image wit provided candiates
+        detector = self.pipeline(images=[image], candidate_labels=inputs["candidate_labels"])
         return detector