lemonaddie
commited on
Commit
•
d35e5ee
1
Parent(s):
db72d60
Update app.py
Browse files
app.py
CHANGED
@@ -24,27 +24,22 @@ def process(
|
|
24 |
path_out_vis=None,
|
25 |
):
|
26 |
|
27 |
-
print('4424')
|
28 |
if path_out_vis is not None:
|
29 |
return (
|
30 |
[path_out_16bit, path_out_vis],
|
31 |
[path_out_16bit, path_out_fp32, path_out_vis],
|
32 |
)
|
33 |
|
34 |
-
print('44a4')
|
35 |
input_image = Image.open(path_input)
|
36 |
-
print('55b5')
|
37 |
|
38 |
-
print('aaa')
|
39 |
pipe_out = pipe(
|
40 |
input_image,
|
41 |
-
ensemble_size=ensemble_size,
|
42 |
denoising_steps=denoise_steps,
|
|
|
43 |
processing_res=processing_res,
|
44 |
batch_size=1 if processing_res == 0 else 0,
|
45 |
show_progress_bar=True,
|
46 |
)
|
47 |
-
print('bbb')
|
48 |
|
49 |
depth_pred = pipe_out.depth_np
|
50 |
depth_colored = pipe_out.depth_colored
|
@@ -102,7 +97,7 @@ def run_demo_server(pipe):
|
|
102 |
type="filepath",
|
103 |
)
|
104 |
with gr.Accordion("Advanced options", open=False):
|
105 |
-
|
106 |
[
|
107 |
("Outdoor", "outdoor"),
|
108 |
("Indoor", "indoor"),
|
@@ -111,7 +106,7 @@ def run_demo_server(pipe):
|
|
111 |
label="Data Domain",
|
112 |
value="indoor",
|
113 |
)
|
114 |
-
|
115 |
label="Classifier Free Guidance Scale",
|
116 |
minimum=1,
|
117 |
maximum=5,
|
@@ -153,7 +148,7 @@ def run_demo_server(pipe):
|
|
153 |
visible=False,
|
154 |
)
|
155 |
with gr.Row():
|
156 |
-
submit_btn = gr.Button(value="Compute
|
157 |
clear_btn = gr.Button(value="Clear")
|
158 |
with gr.Column():
|
159 |
output_slider = ImageSlider(
|
|
|
24 |
path_out_vis=None,
|
25 |
):
|
26 |
|
|
|
27 |
if path_out_vis is not None:
|
28 |
return (
|
29 |
[path_out_16bit, path_out_vis],
|
30 |
[path_out_16bit, path_out_fp32, path_out_vis],
|
31 |
)
|
32 |
|
|
|
33 |
input_image = Image.open(path_input)
|
|
|
34 |
|
|
|
35 |
pipe_out = pipe(
|
36 |
input_image,
|
|
|
37 |
denoising_steps=denoise_steps,
|
38 |
+
ensemble_size=ensemble_size,
|
39 |
processing_res=processing_res,
|
40 |
batch_size=1 if processing_res == 0 else 0,
|
41 |
show_progress_bar=True,
|
42 |
)
|
|
|
43 |
|
44 |
depth_pred = pipe_out.depth_np
|
45 |
depth_colored = pipe_out.depth_colored
|
|
|
97 |
type="filepath",
|
98 |
)
|
99 |
with gr.Accordion("Advanced options", open=False):
|
100 |
+
domain = gr.Radio(
|
101 |
[
|
102 |
("Outdoor", "outdoor"),
|
103 |
("Indoor", "indoor"),
|
|
|
106 |
label="Data Domain",
|
107 |
value="indoor",
|
108 |
)
|
109 |
+
cfg_scale = gr.Slider(
|
110 |
label="Classifier Free Guidance Scale",
|
111 |
minimum=1,
|
112 |
maximum=5,
|
|
|
148 |
visible=False,
|
149 |
)
|
150 |
with gr.Row():
|
151 |
+
submit_btn = gr.Button(value="Compute", variant="primary")
|
152 |
clear_btn = gr.Button(value="Clear")
|
153 |
with gr.Column():
|
154 |
output_slider = ImageSlider(
|