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