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

2c8aa2f

·

verified ·

1 Parent(s): 29b960d

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -38,7 +38,7 @@ class EndpointHandler():
             'scores': results['scores'].tolist(),
             'text_labels': results['text_labels']  # Already serializable
         }
-        return results
         #prediction = self.pipeline(image=[image], candidate_labels=inputs["candidates"])
         #return prediction[0]

             'scores': results['scores'].tolist(),
             'text_labels': results['text_labels']  # Already serializable
         }
+        return serializable_results
         #prediction = self.pipeline(image=[image], candidate_labels=inputs["candidates"])
         #return prediction[0]