Spaces:
Sleeping
Sleeping
Tonioesparza
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -127,7 +127,7 @@ def ourhood_inference(prompt=str,num_inference_steps=int,scaffold=int,seed=int):
|
|
127 |
mask1 = load_image(scaff_dic[scaffold]['mask1'])
|
128 |
mask2 = load_image(scaff_dic[scaffold]['mask2'])
|
129 |
|
130 |
-
masks = processor_mask.preprocess([
|
131 |
masks = [masks.reshape(1, masks.shape[0], masks.shape[2], masks.shape[3])]
|
132 |
|
133 |
###precomputed depth image
|
|
|
127 |
mask1 = load_image(scaff_dic[scaffold]['mask1'])
|
128 |
mask2 = load_image(scaff_dic[scaffold]['mask2'])
|
129 |
|
130 |
+
masks = processor_mask.preprocess([mask1], height=output_height, width=output_width)
|
131 |
masks = [masks.reshape(1, masks.shape[0], masks.shape[2], masks.shape[3])]
|
132 |
|
133 |
###precomputed depth image
|