Spaces:
Runtime error
Runtime error
pablo
commited on
Commit
•
c7d558b
1
Parent(s):
07fe287
my bad
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ else:
|
|
34 |
|
35 |
|
36 |
def estimate_depth(image):
|
37 |
-
|
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:
|