Blueway
/

inference-endpoint-for-omdet-turbo-swin-tiny-hf

Zero-Shot Object Detection

endpoints-template

Model card Files Files and versions Community

LDanielBlueway commited on Mar 19

Commit

5dd94e4

·

verified ·

1 Parent(s): f77b2b9

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

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

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