Update handler.py

handler.py

@@ -33,6 +33,13 @@ class EndpointHandler():
             score_threshold=0.3,
             nms_threshold=0.3,
         )[0]
+        # Convert tensors to lists
+        serializable_results = {
+            'boxes': results['boxes'].tolist(),
+            'scores': results['scores'].tolist(),
+            'labels': results['labels'].tolist(),
+            'text_labels': results['text_labels']  # Already serializable
+        }
         return results
         #prediction = self.pipeline(image=[image], candidate_labels=inputs["candidates"])
         #return prediction[0]