Update app.py

app.py

@@ -43,14 +43,15 @@ def process_image(image_path):
     output = prediction.cpu().numpy()
     depth_image = (output * 255 / np.max(output)).astype("uint8")
     depth_image_padded = np.array(
-        ImageOps.pad(Image.fromarray(depth_image), (1280, 720))
+      Image.fromarray(depth_image)
     )
 
-   
 
+    # Return as downloadable file
     return depth_image_padded
 
 
+
 examples_images = [[f] for f in sorted(glob.glob("examples/*.jpg"))]
 
 
@@ -88,4 +89,4 @@ to generate the autostereogram (Magic Eye)
         inputs=[input_image],
         outputs=predicted_depth,
     )
-blocks.launch(debug=True)
+blocks.launch(debug=True)