pablo commited on
Commit
07fe287
1 Parent(s): aae12bf
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ def predict(dict, depth, prompt="", negative_prompt="", guidance_scale=7.5, step
71
  init_image = dict["image"].convert("RGB").resize((512, 512))
72
 
73
  if (depth == None):
74
- depth_image = estimate_depth(image)
75
  else:
76
  depth_image = depth.convert("L")
77
 
 
71
  init_image = dict["image"].convert("RGB").resize((512, 512))
72
 
73
  if (depth == None):
74
+ depth_image = estimate_depth(init_image)
75
  else:
76
  depth_image = depth.convert("L")
77