Fix saving and returning of output image

app.py

@@ -104,8 +104,11 @@ with gr.Blocks(css=css) as demo:
 
             pred_depth = pred_depth.cpu().numpy().squeeze()
             output_image = plt.imsave('depth.png', pred_depth, cmap='jet')
-
-            return [(original_image, output_image), output_image]
+            tmp = tempfile.NamedTemporaryFile(delete=False)
+            tmp.write(output_image)
+            print(tmp.read())
+            tmp_output = tmp.read()
+            return [(original_image, tmp_output), tmp_output]
 
     submit.click(on_submit, inputs=[input_image], outputs=[
                  depth_image_slider, raw_file])