Update handler.py

handler.py

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