Spaces:
Running
on
Zero
Running
on
Zero
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()
|