radames commited on
Commit
dcacd5b
1 Parent(s): 2265388

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -33,6 +33,7 @@ def predict(input_image, prompt, negative_prompt, steps, num_samples, scale, see
33
  depth = None
34
  if depth_image is not None:
35
  depth_image = pad_image(depth_image)
 
36
  depth = np.array(depth_image.convert("L"))
37
  depth = depth.astype(np.float32) / 255.0
38
  depth = depth[None, None]
 
33
  depth = None
34
  if depth_image is not None:
35
  depth_image = pad_image(depth_image)
36
+ depth_image = depth_image.resize((512, 512))
37
  depth = np.array(depth_image.convert("L"))
38
  depth = depth.astype(np.float32) / 255.0
39
  depth = depth[None, None]