Update app.py

app.py

@@ -102,9 +102,10 @@ with gr.Blocks(css=css) as demo:
 
     with gr.Row():
         input_image = gr.Image(label="Input Image", type='numpy', elem_id='img-display-input')
-        dis_image_slider = ImageSlider(label="Pedict View", elem_id='img-display-output', position=0.5)
+        dis_image_slider = gr.Image(label="Pedict View", type='numpy', elem_id='img-display-input')
+        # dis_image_slider = ImageSlider(label="Pedict View", elem_id='img-display-output', position=0.5)
     submit = gr.Button(value="Compute")
-    raw_file = gr.File(label="16-bit raw output (can be considered as disparity)", elem_id="download",)
+    raw_file = gr.File(label="16-bit raw output", elem_id="download",)
 
     def on_submit(image):
         original_image = image.copy()
@@ -115,7 +116,8 @@ with gr.Blocks(css=css) as demo:
         tmp_raw_DIS_map = tempfile.NamedTemporaryFile(suffix='.png', delete=False)
         raw_DIS_map.save(tmp_raw_DIS_map.name)
 
-        return [[original_image, DIS_map.astype(np.uint8)], tmp_raw_DIS_map.name]
+        # return [[original_image, DIS_map.astype(np.uint8)], tmp_raw_DIS_map.name]
+        return [DIS_map.astype(np.uint8), tmp_raw_DIS_map.name]
 
     submit.click(on_submit, inputs=[input_image], outputs=[dis_image_slider, raw_file])