Update app.py

app.py

@@ -59,7 +59,7 @@ def predict(radio, dict, word_mask, prompt=""):
         mask = Image.fromarray(np.uint8(bw_image)).convert('RGB')
         os.remove(filename)
     #with autocast("cuda"):
-    output = pipe(prompt = prompt, init_image=init_image, mask_image=mask, strength=0.8)
+    output = pipe(prompt = prompt, image=init_image, mask_image=mask, strength=0.8)
     return output.images[0]
 
 # examples = [[dict(image="init_image.png", mask="mask_image.png"), "A panda sitting on a bench"]]