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:
|