Spaces:
Sleeping
Sleeping
Update model.py
Browse files
model.py
CHANGED
@@ -244,7 +244,8 @@ class Model:
|
|
244 |
lambda_: float,
|
245 |
num_steps: int,
|
246 |
) -> PIL.Image.Image:
|
247 |
-
data = self.prepare_data(image['image'], image['mask'], text)
|
|
|
248 |
|
249 |
uncond_input = self.tokenizer(
|
250 |
[''] * data['pixel_values'].shape[0],
|
|
|
244 |
lambda_: float,
|
245 |
num_steps: int,
|
246 |
) -> PIL.Image.Image:
|
247 |
+
# data = self.prepare_data(image['image'], image['mask'], text)
|
248 |
+
data = self.prepare_data(image['background'], image['layers'][0], text)
|
249 |
|
250 |
uncond_input = self.tokenizer(
|
251 |
[''] * data['pixel_values'].shape[0],
|