Update handler.py

handler.py

@@ -20,15 +20,13 @@ class EndpointHandler:
         img = Image.open(BytesIO(decoded_image)).convert("RGB")
         return img
 
-    def handle(self, event, context):
+    def __call__(self, data, context):
         """Handle the incoming request."""
         try:
             # Extract the inputs from the event
-            data = event['body']
             inputs = data.pop("inputs", data)
-            print(inputs)
             input_image = inputs['image']
-            question = inputs['question']
+            question = inputs.get('question', "move to the red ball")
 
             # Preprocess the image
             img = self.preprocess_image(input_image)