Update app.py

app.py

@@ -100,7 +100,8 @@ with gr.Blocks(css=css) as demo:
     raw_file = gr.File(label="16-bit raw output (can be considered as disparity)", elem_id="download",)
     cmap = matplotlib.colormaps.get_cmap('Spectral_r')
 
-    def on_submit(image):
+    def on_submit(img_d):
+        image = cv2.cvtColor(img_d["composite"], cv2.COLOR_RGBA2RGB)
         original_image = image.copy()
 
         h, w = image.shape[:2]