Update app.py

app.py

@@ -82,14 +82,8 @@ def save_splat_file(splat_data, output_path):
     with open(output_path, "wb") as f:
         f.write(splat_data)
 
-def get_reconstructed_scene(outdir, model, device):
+def get_reconstructed_scene(outdir, image_files, model, device):
 
-    image_files = sorted(
-        [
-            os.path.join(outdir, "images", f)
-            for f in os.listdir(os.path.join(outdir, "images"))
-        ]
-    )
     images = [process_image(img_path) for img_path in image_files]
     images = torch.stack(images, dim=0).unsqueeze(0).to(device)  # [1, K, 3, 448, 448]
     b, v, c, h, w = images.shape
@@ -270,7 +264,7 @@ def generate_splats_from_images(image_paths, session_id=None):
 
     print("Running run_model...")
     with torch.no_grad():
-        plyfile, video, depth_colored = get_reconstructed_scene(base_dir, model, device)
+        plyfile, video, depth_colored = get_reconstructed_scene(base_dir, image_paths, model, device)
 
     end_time = time.time()
     print(f"Total time: {end_time - start_time:.2f} seconds (including IO)")