Update handler.py

handler.py

@@ -39,7 +39,7 @@ class EndpointHandler:
         init_image.thumbnail((width, height))
 
         generator = torch.Generator(device="cuda").manual_seed(seed)
-        images = self.pipe(prompts, init_image=init_image,generator = generator, **data).images
+        images = self.pipe(prompts, image=init_image,generator = generator, **data).images
         img_strs = []
 
         for image in images: