alexnasa commited on
Commit
a19ea4b
·
verified ·
1 Parent(s): 852e884

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -153,10 +153,9 @@ def generate_splats_from_video(video_path, session_id=None):
153
  if session_id is None:
154
  session_id = uuid.uuid4().hex
155
 
156
- # unpack the 3 outputs from the helper
157
  plyfile, rgb_vid, depth_vid = generate_splats_from_images(images_folder, session_id)
158
 
159
- # return all four in order
160
  return plyfile, rgb_vid, depth_vid, image_paths
161
 
162
  @spaces.GPU()
 
153
  if session_id is None:
154
  session_id = uuid.uuid4().hex
155
 
156
+ images_folder, image_paths = extract_frames(video_path, session_id)
157
  plyfile, rgb_vid, depth_vid = generate_splats_from_images(images_folder, session_id)
158
 
 
159
  return plyfile, rgb_vid, depth_vid, image_paths
160
 
161
  @spaces.GPU()