Update app.py

app.py

@@ -45,20 +45,11 @@ def process_image(image_path):
     ).squeeze()
     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))
+    depth_image_padded = Image.fromarray(depth_image)
     )
 
-    stereo_image = stereo_converter.convert_depth_to_stereogram_with_thread_pool(
-        depth_image_padded, False
-    ).astype(np.uint8)
 
-    stereo_image_pil = Image.fromarray(stereo_image).convert("RGB")
-    with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as f:
-        image_name = f.name
-        stereo_image_pil.save(image_name)
-
-    return [depth_image_padded, stereo_image, image_name]
+    return [depth_image_padded]
 
 
 examples_images = [[f] for f in sorted(glob.glob("examples/*.jpg"))]
@@ -90,12 +81,12 @@ to generate the autostereogram (Magic Eye)
             examples=examples_images,
             fn=process_image,
             inputs=[input_image],
-            outputs=[predicted_depth, autostereogram, file_download],
+            outputs=[predicted_depth],
             cache_examples=True,
         )
     button.click(
         fn=process_image,
         inputs=[input_image],
-        outputs=[predicted_depth, autostereogram, file_download],
+        outputs=[predicted_depth],
     )
 blocks.launch(debug=True)