RaynWu2002 commited on
Commit
8e2646d
·
verified ·
1 Parent(s): 2bf18ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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()