Add fix for 3DGS tab

demo/gs_demo.py

@@ -2,7 +2,7 @@ import gradio as gr
 from gs_train import train
 import os
 
-DATASET_DIR = "colmap_data"
+from demo_globals import CACHE_PATH, MODEL, DEVICE, SILENT, DATASET_DIR
 
 def get_dataset_folders(datasets_path):
     try:
@@ -10,9 +10,9 @@ def get_dataset_folders(datasets_path):
     except FileNotFoundError:
         return []
 
-def gs_demo_tab(cache_path):
+def gs_demo_tab():
     # datasets_path = "/app/data/scenes/"
-    dataset_path = os.path.join(cache_path, DATASET_DIR)
+    dataset_path = os.path.join(CACHE_PATH, DATASET_DIR)
     def start_training(selected_folder, *args):
         selected_data_path = os.path.join(datasets_path, selected_folder)
         return train(selected_data_path, *args)

demo/gs_train.py

@@ -60,7 +60,7 @@ class TrainingArgs:
     checkpoint_iterations: list[int] = field(default_factory=lambda: [7_000, 15_000, 30_000])
     start_checkpoint: str = None
 
-@spaces.GPU(duration=20)
+@spaces.GPU(duration=10)
 def train(
     data_source_path, sh_degree, model_path, images, resolution, white_background, data_device, eval,
     convert_SHs_python, compute_cov3D_python, debug,