Update handler.py

handler.py

@@ -12,9 +12,10 @@ class EndpointHandler():
         self.pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(self.pipe.scheduler.config)
     
     def __call__(self, data):
-        
-        image=data.pop("image",data)
-        prompt=data.pop("prompt",data).decode('utf-8')
+
+        info=data['inputs']
+        image=info.pop("image",data)
+        prompt=info.pop("prompt",data).decode('utf-8')
         image=base64.b64decode(image)
         raw_images = Image.open(BytesIO(image)).convert('RGB')