Update model.py

model.py

@@ -244,7 +244,8 @@ class Model:
         lambda_: float,
         num_steps: int,
     ) -> PIL.Image.Image:
-        data = self.prepare_data(image['image'], image['mask'], text)
+        # data = self.prepare_data(image['image'], image['mask'], text)
+        data = self.prepare_data(image['background'], image['layers'][0], text)
 
         uncond_input = self.tokenizer(
             [''] * data['pixel_values'].shape[0],