修复img2img的报错 (#4)

- 修复img2img的报错 (ec8d02b4e0cd0d97a496d618d7b69be6eb48de26)


Co-authored-by: HaloMaster <HaloMaster@users.noreply.huggingface.co>

app.py

@@ -25,7 +25,7 @@ pipe_img2img = StableDiffusionImg2ImgPipeline(**pipe_text2img.components).to(dev
 def infer_text2img(prompt, guide, steps, width, height, image_in, strength):
     if image_in is not None:
         init_image = image_in.convert("RGB").resize((width, height))
-        output = pipe_img2img(prompt, init_image=init_image, strength=strength, width=width, height=height, guidance_scale=guide, num_inference_steps=steps)
+        output = pipe_img2img(prompt, image=init_image, strength=strength, guidance_scale=guide, num_inference_steps=steps)
     else:
         output = pipe_text2img(prompt, width=width, height=height, guidance_scale=guide, num_inference_steps=steps,)
     image = output.images[0]