Update core.py

core.py

@@ -51,7 +51,7 @@ def get_config():
 
 
 # Infer
-@spaces.GPU
+@spaces.GPU(duration=120)
 def run_infer_script(
     pitch: int,
     filter_radius: int,
@@ -103,7 +103,7 @@ def run_infer_script(
 
 
 # Batch infer
-@spaces.GPU
+@spaces.GPU(duration=200)
 def run_batch_infer_script(
     pitch: int,
     filter_radius: int,
@@ -170,7 +170,7 @@ def run_batch_infer_script(
 
 
 # TTS
-@spaces.GPU
+@spaces.GPU(duration=120)
 def run_tts_script(
     tts_text: str,
     tts_voice: str,
@@ -246,7 +246,7 @@ def run_tts_script(
 
 
 # Preprocess
-@spaces.GPU()
+@spaces.GPU(duration=120)
 def run_preprocess_script(
     model_name: str, dataset_path: str, sample_rate: int, cpu_cores: int
 ):
@@ -273,7 +273,7 @@ def run_preprocess_script(
 
 
 # Extract
-@spaces.GPU
+@spaces.GPU(duration=120)
 def run_extract_script(
     model_name: str,
     rvc_version: str,
@@ -331,7 +331,7 @@ def run_extract_script(
 
 
 # Train
-@spaces.GPU()
+@spaces.GPU(duration=360)
 def run_train_script(
     model_name: str,
     rvc_version: str,