Update image saving method

app.py

@@ -115,7 +115,8 @@ with gr.Blocks(css=css) as demo:
 
             # Continue with your file saving operations
             tmp = tempfile.NamedTemporaryFile(suffix='.png', delete=False)
-            cv2.imwrite(tmp.name, output_image)
+            # cv2.imwrite(tmp.name, output_image)
+            plt.imsave(tmp.name, pred_depth, cmap='jet')
 
             return [(original_image, output_image), tmp.name]