Update app.py

app.py

@@ -38,7 +38,7 @@ def gen_fill_api_call(image_base64_file, mask_base64_file, prompt):
     return res_image
 
 
-def predict(dict):
+def predict(dict, prompt):
 
     init_image = Image.fromarray(dict['background'][:, :, :3], 'RGB') #dict['background'].convert("RGB")#.resize((1024, 1024))
     mask = Image.fromarray(dict['layers'][0][:,:,3], 'L') #dict['layers'].convert("RGB")#.resize((1024, 1024))