adpro commited on
Commit
cc376eb
·
verified ·
1 Parent(s): 68aeb60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -11
app.py CHANGED
@@ -49,16 +49,9 @@ def process_image(image_path):
49
  ImageOps.pad(Image.fromarray(depth_image), (1280, 720))
50
  )
51
 
52
- stereo_image = stereo_converter.convert_depth_to_stereogram_with_thread_pool(
53
- depth_image_padded, False
54
- ).astype(np.uint8)
55
 
56
- stereo_image_pil = Image.fromarray(stereo_image).convert("RGB")
57
- with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as f:
58
- image_name = f.name
59
- stereo_image_pil.save(image_name)
60
-
61
- return [depth_image_padded, stereo_image, image_name]
62
 
63
 
64
  examples_images = [[f] for f in sorted(glob.glob("examples/*.jpg"))]
@@ -90,12 +83,12 @@ to generate the autostereogram (Magic Eye)
90
  examples=examples_images,
91
  fn=process_image,
92
  inputs=[input_image],
93
- outputs=[predicted_depth, autostereogram, file_download],
94
  cache_examples=True,
95
  )
96
  button.click(
97
  fn=process_image,
98
  inputs=[input_image],
99
- outputs=[predicted_depth, autostereogram, file_download],
100
  )
101
  blocks.launch(debug=True)
 
49
  ImageOps.pad(Image.fromarray(depth_image), (1280, 720))
50
  )
51
 
52
+
 
 
53
 
54
+ return depth_image_padded
 
 
 
 
 
55
 
56
 
57
  examples_images = [[f] for f in sorted(glob.glob("examples/*.jpg"))]
 
83
  examples=examples_images,
84
  fn=process_image,
85
  inputs=[input_image],
86
+ outputs=predicted_depth],
87
  cache_examples=True,
88
  )
89
  button.click(
90
  fn=process_image,
91
  inputs=[input_image],
92
+ outputs=predicted_depth],
93
  )
94
  blocks.launch(debug=True)