Update app.py

app.py

@@ -76,8 +76,8 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=5, steps=30, str
         negative_prompt = None
 
     
-    init_image = dict["image"].convert("RGB")#.resize((1024, 1024))
-    mask = dict["mask"].convert("RGB")#.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))
 
     w,h = get_size(init_image)