Fix image saving bug

app.py

@@ -104,12 +104,8 @@ with gr.Blocks(css=css) as demo:
             print("== Finished processing image")
             pred_depth = pred_depth.cpu().numpy().squeeze()
             output_image = plt.imsave('depth.png', pred_depth, cmap='jet')
-            print("== Saved image"+str(output_image))
-            tmp = tempfile.NamedTemporaryFile(delete=False)
-            output_image.save(tmp.name)
-            print("== Saved image"+str(tmp.name))
 
-            return [(original_image, output_image), output_image]
+            return [(original_image, original_image), output_image]
 
     submit.click(on_submit, inputs=[input_image], outputs=[
                  depth_image_slider, raw_file])