Extend ZeroGPU duration for dataset runs

app.py

@@ -65,6 +65,21 @@ def log_progress(message: str, level: int = 1, enabled: bool = True) -> None:
 # Load environment variables
 load_dotenv()
 
+_GPU_DURATION = os.getenv("CHIZZLER_GPU_DURATION")
+if _GPU_DURATION is not None:
+    DEFAULT_GPU_DURATION = int(_GPU_DURATION)
+else:
+    DEFAULT_GPU_DURATION = 3600 if os.getenv("SPACE_ID") else 0
+
+
+def gpu_decorator(duration: int):
+    if duration and duration > 0:
+        try:
+            return spaces.GPU(duration=duration)
+        except TypeError:
+            return spaces.GPU
+    return spaces.GPU
+
 
 def get_hf_token() -> Optional[str]:
     return (
@@ -722,7 +737,7 @@ def default_output_repo(source_id: str, username: str) -> str:
     return f"{username}/{name}"
 
 
-@spaces.GPU
+@gpu_decorator(DEFAULT_GPU_DURATION)
 def process_dataset_and_push(
     dataset_id: str,
     config: str,
@@ -1206,7 +1221,7 @@ def assemble_cached_dataset_and_push(
     )
 
 
-@spaces.GPU
+@gpu_decorator(DEFAULT_GPU_DURATION)
 def gradio_single_file(
     audio_file,
     vad_threshold,