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()
|