Spaces:
Runtime error
Runtime error
lemonaddie
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -276,28 +276,7 @@ def main():
|
|
276 |
|
277 |
pipe = pipe.to(device)
|
278 |
|
279 |
-
|
280 |
-
|
281 |
-
pipe_out = pipe(
|
282 |
-
input_image,
|
283 |
-
denoising_steps=5,
|
284 |
-
ensemble_size=1,
|
285 |
-
processing_res=768,
|
286 |
-
batch_size=0,
|
287 |
-
guidance_scale=3,
|
288 |
-
domain="indoor",
|
289 |
-
show_progress_bar=True,
|
290 |
-
)
|
291 |
-
|
292 |
-
depth_pred: np.ndarray = pipe_out.depth_np
|
293 |
-
depth_colored: Image.Image = pipe_out.depth_colored
|
294 |
-
normal_pred: np.ndarray = pipe_out.normal_np
|
295 |
-
normal_colored: Image.Image = pipe_out.normal_colored
|
296 |
-
|
297 |
-
print(depth_pred.shape)
|
298 |
-
print(np.unique(depth_pred))
|
299 |
-
|
300 |
-
# run_demo_server(pipe)
|
301 |
|
302 |
|
303 |
if __name__ == "__main__":
|
|
|
276 |
|
277 |
pipe = pipe.to(device)
|
278 |
|
279 |
+
run_demo_server(pipe)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
280 |
|
281 |
|
282 |
if __name__ == "__main__":
|