Thomasboosinger
/

owlvit-base-patch32

Zero-Shot Object Detection

Model card Files Files and versions

Thomasboosinger commited on Feb 23, 2024

Commit

0606046

·

verified ·

1 Parent(s): e0734d3

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -27,6 +27,6 @@ class EndpointHandler:
         image = Image.open(BytesIO(base64.b64decode(data['inputs']['image'])))
         # Run detection and obtain results.
-        results = self.pipeline(image=image, candidate_labels=data['inputs']['candidates'])
         return results

         image = Image.open(BytesIO(base64.b64decode(data['inputs']['image'])))
         # Run detection and obtain results.
+        results = self.pipeline(image=image, candidate_labels=data['inputs']['candidates'], threshold = .01)
         return results