Tonioesparza commited on
Commit
dfcae18
·
verified ·
1 Parent(s): be9621a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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([mask2], 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
 
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