Update core.py

core.py

@@ -28,7 +28,7 @@ locales = list({voice["Locale"] for voice in voices_data})
 
 
 # Infer
-@spaces.GPU(duration=120)
+@spaces.GPU
 def run_infer_script(
     f0up_key,
     filter_radius,
@@ -80,6 +80,7 @@ def run_infer_script(
 
 
 # Batch infer
+@spaces.GPU
 def run_batch_infer_script(
     f0up_key,
     filter_radius,
@@ -147,6 +148,7 @@ def run_batch_infer_script(
 
 
 # TTS
+@spaces.GPU
 def run_tts_script(
     tts_text,
     tts_voice,
@@ -227,20 +229,6 @@ def run_prerequisites_script(pretraineds_v1, pretraineds_v2, models, exe):
     prequisites_download_pipeline(pretraineds_v1, pretraineds_v2, models, exe)
     return "Prerequisites installed successfully."
 
-
-# API
-def run_api_script(ip, port):
-    command = [
-        "env/Scripts/uvicorn.exe" if os.name == "nt" else "uvicorn",
-        "api:app",
-        "--host",
-        ip,
-        "--port",
-        port,
-    ]
-    subprocess.run(command)
-
-
 # Parse arguments
 def parse_arguments():
     parser = argparse.ArgumentParser(
@@ -694,13 +682,6 @@ def parse_arguments():
         help="Download executables",
     )
 
-    # Parser for 'api' mode
-    api_parser = subparsers.add_parser("api", help="Run the API")
-    api_parser.add_argument(
-        "--host", type=str, help="Host address", default="127.0.0.1"
-    )
-    api_parser.add_argument("--port", type=str, help="Port", default="8000")
-
     return parser.parse_args()
 
 
@@ -792,11 +773,6 @@ def main():
                 str(args.models),
                 str(args.exe),
             )
-        elif args.mode == "api":
-            run_api_script(
-                str(args.host),
-                str(args.port),
-            )
     except Exception as error:
         print(f"Error: {error}")