Updated handler file

handler.py

@@ -38,9 +38,12 @@ class EndpointHandler():
             A :obj:`dict`:. The object returned should be a dict of one list like {"captions": ["A hugging face at the office"]} containing :
                 - "caption": A string corresponding to the generated caption.
         """
-        img_data = data.pop("image", data)
+        print("********* Helllo ***********")
+        print(data)
+        img_data = data.pop("input", data)
         prompt = data.pop("prompt", None)
-        parameters = data.pop("parameters", {})
+        print("#########")
+        # parameters = data.pop("parameters", {})
 
         if isinstance(img_data, Image.Image):
             raw_image = img_data