freealise commited on
Commit
1072e96
1 Parent(s): d3af6ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -177,7 +177,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
177
  frames = orig_frames
178
  depths = depth_frames
179
  masks = orig_frames
180
- return final_vid, final_zip, gr.Gallery(value=frames), masks[frame_selected] #output_path
181
 
182
  def depth_edges_mask(depth):
183
  """Returns a mask of edges in the depth map.
@@ -327,9 +327,11 @@ def switch_rows(v):
327
  global frames
328
  global depths
329
  if v == True:
330
- return gr.Gallery(value=depths)
 
331
  else:
332
- return gr.Gallery(value=frames)
 
333
 
334
 
335
  css = """
 
177
  frames = orig_frames
178
  depths = depth_frames
179
  masks = orig_frames
180
+ return final_vid, final_zip, frames, masks[frame_selected] #output_path
181
 
182
  def depth_edges_mask(depth):
183
  """Returns a mask of edges in the depth map.
 
327
  global frames
328
  global depths
329
  if v == True:
330
+ print(depths[0])
331
+ return depths
332
  else:
333
+ print(frames[0])
334
+ return frames
335
 
336
 
337
  css = """