Update handler.py

handler.py

@@ -33,10 +33,10 @@ class EndpointHandler():
         
 
     def __call__(self, event):
-        if "url" not in event:
+        if "inputs" not in event:
             return {
                 "statusCode": 400,
-                "body": json.dumps("Error: Please provide an 'url' parameter."),
+                "body": json.dumps("Error: Please provide an 'inputs' parameter."),
             }
 
         if "prompt" not in event:
@@ -45,8 +45,9 @@ class EndpointHandler():
                 "body": json.dumps("Error: Please provide an 'prompt' parameter."),
             }    
 
-        url = event["url"]
-        prompt = event["prompt"]
+        inputs = event["inputs"]
+        url = inputs["url"]
+        prompt = inputs["prompt"]
         parsed_answer = self.predict_image(self, url, prompt)
 
         return {