pablo commited on
Commit
c7d558b
1 Parent(s): 07fe287
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,7 +34,7 @@ else:
34
 
35
 
36
  def estimate_depth(image):
37
- image= image.resize((384,384)).convert("RGB")
38
  image = np.array(image)
39
  input_batch = transform(image).to(device)
40
 
@@ -69,7 +69,7 @@ def predict(dict, depth, prompt="", negative_prompt="", guidance_scale=7.5, step
69
  scheduler_class_name = scheduler.split("-")[0]
70
 
71
  init_image = dict["image"].convert("RGB").resize((512, 512))
72
-
73
  if (depth == None):
74
  depth_image = estimate_depth(init_image)
75
  else:
 
34
 
35
 
36
  def estimate_depth(image):
37
+
38
  image = np.array(image)
39
  input_batch = transform(image).to(device)
40
 
 
69
  scheduler_class_name = scheduler.split("-")[0]
70
 
71
  init_image = dict["image"].convert("RGB").resize((512, 512))
72
+
73
  if (depth == None):
74
  depth_image = estimate_depth(init_image)
75
  else: