Fix color mapping issue in app.py

app.py

@@ -103,8 +103,9 @@ with gr.Blocks(css=css) as demo:
                 pred_depths_r_list[-1], pred_depths_r_list_flipped[-1])
             print("== Finished processing image")
             pred_depth = pred_depth.cpu().numpy().squeeze()
+            pred_outpu_depth = pred_depth.cpu().numpy().astype(np.uint8)
             output_image = cv2.applyColorMap(
-                pred_depth, cv2.COLORMAP_INFERNO)[:, :, ::-1]
+                pred_outpu_depth, cv2.COLORMAP_INFERNO)[:, :, ::-1]
 
             # Continue with your file saving operations
             tmp = tempfile.NamedTemporaryFile(suffix='.png', delete=False)