Update handler.py

handler.py

@@ -30,13 +30,13 @@ class EndpointHandler():
         return predictions_array
 
     def __call__(self, event):
-        if "image_path" not in event:
+        if "inputs" not in event:
             return {
                 "statusCode": 400,
                 "body": json.dumps("Error: Please provide an 'image_path' parameter."),
             }
 
-        image_path = event["image_path"]
+        image_path = event["inputs"]
 
         try:
             predictions = self.predict_objects(image_path)