Update handler.py

handler.py

@@ -58,7 +58,7 @@ class EndpointHandler:
         # 3. Run the img2img pipeline
         result = pipe(
             prompt=prompt,
-            init_image=init_img,
+            image=init_img,
             strength=strength,
             guidance_scale=guidance_scale,
             num_inference_steps=num_steps,