Update app.py

app.py

@@ -45,11 +45,20 @@ def process_image(image_path):
     ).squeeze()
     output = prediction.cpu().numpy()
     depth_image = (output * 255 / np.max(output)).astype("uint8")
-    depth_image_padded = Image.fromarray(depth_image)
-    
+    depth_image_padded = np.array(
+        ImageOps.pad(Image.fromarray(depth_image), (1280, 720))
+    )
+
+    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,]
+    return [depth_image_padded, stereo_image, image_name]
 
 
 examples_images = [[f] for f in sorted(glob.glob("examples/*.jpg"))]
@@ -81,12 +90,12 @@ to generate the autostereogram (Magic Eye)
             examples=examples_images,
             fn=process_image,
             inputs=[input_image],
-            outputs=[predicted_depth],
+            outputs=[predicted_depth, autostereogram, file_download],
             cache_examples=True,
         )
     button.click(
         fn=process_image,
         inputs=[input_image],
-        outputs=[predicted_depth,],
+        outputs=[predicted_depth, autostereogram, file_download],
     )
-blocks.launch(debug=True)
+blocks.launch(debug=True)