Refactor depth image saving and display

This commit refactors the code for saving and displaying depth images. It replaces the previous method of saving the predicted depth image with a new method that saves the raw depth image. Additionally, it applies a color map to the depth image for better visualization. This improves the overall functionality and user experience of the application.

app.py

@@ -103,10 +103,18 @@ with gr.Blocks(css=css) as demo:
                 pred_depths_r_list[-1], pred_depths_r_list_flipped[-1])
 
             pred_depth = pred_depth.cpu().numpy().squeeze()
+
+            depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0
+            depth = pred_depth.cpu().numpy().astype(np.uint8)
+            colored_depth = cv2.applyColorMap(
+                depth, cv2.COLORMAP_INFERNO)[:, :, ::-1]
+
+            raw_depth = Image.fromarray(
+                pred_depth.cpu().numpy().astype('uint16'))
             tmp = tempfile.NamedTemporaryFile(suffix='.png', delete=False)
-            pred_depth.save(tmp.name)
+            raw_depth.save(tmp.name)
 
-            return [(original_image, pred_depth), tmp.name]
+            return [(original_image, colored_depth), tmp.name]
 
     submit.click(on_submit, inputs=[input_image], outputs=[
                  depth_image_slider, raw_file])