jens commited on
Commit
b5815d9
·
1 Parent(s): edc462d
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -12,13 +12,15 @@ def snap(image, video):
12
  gltf_path = create_3d_obj(np.array(image), depth_result)
13
  #segment_predictor = SegmentPredictor()
14
  #sam_result = segment_predictor.predict(image)
15
- return [gltf_path, video]#[depth_result, gltf_path, gltf_path]
16
 
17
 
18
  demo = gr.Interface(
19
  snap,
20
- [gr.Image(source="webcam", tool=None, type="pil"), gr.Video(source="webcam")],
21
- ["image", "video"],
 
 
22
  )
23
 
24
  if __name__ == "__main__":
 
12
  gltf_path = create_3d_obj(np.array(image), depth_result)
13
  #segment_predictor = SegmentPredictor()
14
  #sam_result = segment_predictor.predict(image)
15
+ return [image, gltf_path]#[depth_result, gltf_path, gltf_path]
16
 
17
 
18
  demo = gr.Interface(
19
  snap,
20
+ inputs=[gr.Image(source="webcam", tool=None, type="pil"),
21
+ gr.Video(source="webcam")],
22
+ outputs=[gr.Image(label="predicted depth", type="pil"),
23
+ gr.Model3D(label="3d mesh reconstruction", clear_color=[1.0, 1.0, 1.0, 1.0]), gr.File(label="3d gLTF")]
24
  )
25
 
26
  if __name__ == "__main__":