Update app.py
Browse files
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 |
-
|
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()
|