Upload folder using huggingface_hub

handler.py

@@ -2,7 +2,7 @@ from typing import Dict, Any, List, Generator
 import torch
 import os
 import logging
-from s2s_pipeline import main, rename_args, parse_arguments, setup_logger, initialize_queues_and_events, build_pipeline
+from s2s_pipeline import main, rename_args, get_default_arguments, setup_logger, initialize_queues_and_events, build_pipeline
 import numpy as np
 from queue import Queue
 import threading
@@ -21,8 +21,7 @@ class EndpointHandler:
             self.parler_tts_handler_kwargs,
             self.melo_tts_handler_kwargs,
             self.chat_tts_handler_kwargs,
-        ) = parse_arguments()
-
+        ) = get_default_arguments()
         setup_logger(self.module_kwargs.log_level)
 
         rename_args(self.whisper_stt_handler_kwargs, "stt")

s2s_pipeline.py

@@ -65,6 +65,20 @@ def rename_args(args, prefix):
 
     args.__dict__["gen_kwargs"] = gen_kwargs
 
+def get_default_arguments():
+    return (
+        ModuleArguments(),
+        SocketReceiverArguments(),
+        SocketSenderArguments(),
+        VADHandlerArguments(),
+        WhisperSTTHandlerArguments(),
+        ParaformerSTTHandlerArguments(),
+        LanguageModelHandlerArguments(),
+        MLXLanguageModelHandlerArguments(),
+        ParlerTTSHandlerArguments(),
+        MeloTTSHandlerArguments(),
+        ChatTTSHandlerArguments(),
+    )
 
 def parse_arguments():
     parser = HfArgumentParser(