Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -131,7 +131,7 @@ with gr.Blocks(css="""
|
|
131 |
rgb_input = gr.Image(label="RGB Image", type="pil")
|
132 |
lr_input = gr.Image(label="Low-res Depth", type="pil", image_mode="I")
|
133 |
with gr.Column():
|
134 |
-
output1 = gr.Image(label="DORNet Output", type="pil", elem_classes=["output-image"])
|
135 |
output2 = gr.Image(label="Normalized Output", type="pil", elem_classes=["output-image"])
|
136 |
|
137 |
model_selector = gr.Dropdown(choices=["RGB-D-D", "TOFDSR"], label="Model Type", value="RGB-D-D")
|
@@ -143,10 +143,12 @@ with gr.Blocks(css="""
|
|
143 |
["examples/TOFDSR/2020_09_08_13_59_59_435_rgb_rgb_crop.png", "examples/TOFDSR/2020_09_08_13_59_59_435_rgb_depth_crop_fill.png", "TOFDSR"],
|
144 |
],
|
145 |
inputs=[rgb_input, lr_input, model_selector],
|
146 |
-
outputs=[output1, output2],
|
|
|
147 |
label="Try Examples ↓"
|
148 |
)
|
149 |
|
150 |
-
run_button.click(fn=infer, inputs=[rgb_input, lr_input, model_selector], outputs=[output1, output2])
|
|
|
151 |
|
152 |
demo.launch()
|
|
|
131 |
rgb_input = gr.Image(label="RGB Image", type="pil")
|
132 |
lr_input = gr.Image(label="Low-res Depth", type="pil", image_mode="I")
|
133 |
with gr.Column():
|
134 |
+
# output1 = gr.Image(label="DORNet Output", type="pil", elem_classes=["output-image"])
|
135 |
output2 = gr.Image(label="Normalized Output", type="pil", elem_classes=["output-image"])
|
136 |
|
137 |
model_selector = gr.Dropdown(choices=["RGB-D-D", "TOFDSR"], label="Model Type", value="RGB-D-D")
|
|
|
143 |
["examples/TOFDSR/2020_09_08_13_59_59_435_rgb_rgb_crop.png", "examples/TOFDSR/2020_09_08_13_59_59_435_rgb_depth_crop_fill.png", "TOFDSR"],
|
144 |
],
|
145 |
inputs=[rgb_input, lr_input, model_selector],
|
146 |
+
# outputs=[output1, output2],
|
147 |
+
outputs=[output2],
|
148 |
label="Try Examples ↓"
|
149 |
)
|
150 |
|
151 |
+
# run_button.click(fn=infer, inputs=[rgb_input, lr_input, model_selector], outputs=[output1, output2])
|
152 |
+
run_button.click(fn=infer, inputs=[rgb_input, lr_input, model_selector], outputs=[output2])
|
153 |
|
154 |
demo.launch()
|