Refactor image processing code

app.py

@@ -95,20 +95,21 @@ with gr.Blocks(css=css) as demo:
         # image = torch.from_numpy(image).unsqueeze(0)
         with torch.no_grad():
             image = torch.autograd.Variable(image.unsqueeze(0))
-
+            print("== Processing image")
             pred_depths_r_list, _, _ = model(image)
             image_flipped = flip_lr(image)
             pred_depths_r_list_flipped, _, _ = model(image_flipped)
             pred_depth = post_process_depth(
                 pred_depths_r_list[-1], pred_depths_r_list_flipped[-1])
-
+            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)
-            tmp.write(output_image)
-            print(tmp.read())
-            tmp_output = tmp.read()
-            return [(original_image, tmp_output), tmp_output]
+            output_image.save(tmp.name)
+            print("== Saved image"+str(tmp.name))
+
+            return [(original_image, output_image), output_image]
 
     submit.click(on_submit, inputs=[input_image], outputs=[
                  depth_image_slider, raw_file])