Update handler.py

handler.py

@@ -23,10 +23,11 @@ class EndpointHandler:
     def handle(self, event, context):
         """Handle the incoming request."""
         try:
-            # Extract the base64-encoded image and question from the event
-            inputs = event.get('inputs', {})
-            input_image = inputs.get('image', '')
-            question = inputs.get('question', "move to the red ball")
+            # Extract the inputs from the event
+            data = event['body']
+            inputs = data.pop("inputs", data)
+            input_image = inputs['image']
+            question = inputs['question']
 
             # Preprocess the image
             img = self.preprocess_image(input_image)