Update app.py

app.py

@@ -111,8 +111,9 @@ def predict(dict, prompt="", negative_prompt = default_negative_prompt, guidance
     if negative_prompt == "":
         negative_prompt = None
 
-    init_image = dict["image"].convert("RGB")#.resize((1024, 1024))
-    mask = dict["mask"].convert("L")#.resize((1024, 1024))
+    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))
+
 
     width, height = get_size(init_image)