andito HF staff commited on
Commit
967aebb
1 Parent(s): c72e80d

Upload folder using huggingface_hub

Browse files
STT/__pycache__/whisper_stt_handler.cpython-311.pyc CHANGED
Binary files a/STT/__pycache__/whisper_stt_handler.cpython-311.pyc and b/STT/__pycache__/whisper_stt_handler.cpython-311.pyc differ
 
__pycache__/baseHandler.cpython-311.pyc ADDED
Binary file (3.74 kB). View file
 
__pycache__/handler.cpython-311.pyc ADDED
Binary file (5.33 kB). View file
 
__pycache__/s2s_pipeline.cpython-311.pyc ADDED
Binary file (15.5 kB). View file
 
handler.py CHANGED
@@ -1,8 +1,8 @@
1
- from typing import Dict, Any, List
2
  import torch
3
  import os
4
  import logging
5
- from s2s_pipeline import main, prepare_args, parse_arguments, setup_logger, initialize_queues_and_events, build_pipeline
6
  import numpy as np
7
  from queue import Queue
8
  import threading
@@ -25,13 +25,13 @@ class EndpointHandler:
25
 
26
  setup_logger(self.module_kwargs.log_level)
27
 
28
- prepare_args(self.whisper_stt_handler_kwargs, "stt")
29
- prepare_args(self.paraformer_stt_handler_kwargs, "paraformer_stt")
30
- prepare_args(self.language_model_handler_kwargs, "lm")
31
- prepare_args(self.mlx_language_model_handler_kwargs, "mlx_lm")
32
- prepare_args(self.parler_tts_handler_kwargs, "tts")
33
- prepare_args(self.melo_tts_handler_kwargs, "melo")
34
- prepare_args(self.chat_tts_handler_kwargs, "chat_tts")
35
 
36
  self.queues_and_events = initialize_queues_and_events()
37
 
 
1
+ from typing import Dict, Any, List, Generator
2
  import torch
3
  import os
4
  import logging
5
+ from s2s_pipeline import main, rename_args, parse_arguments, setup_logger, initialize_queues_and_events, build_pipeline
6
  import numpy as np
7
  from queue import Queue
8
  import threading
 
25
 
26
  setup_logger(self.module_kwargs.log_level)
27
 
28
+ rename_args(self.whisper_stt_handler_kwargs, "stt")
29
+ rename_args(self.paraformer_stt_handler_kwargs, "paraformer_stt")
30
+ rename_args(self.language_model_handler_kwargs, "lm")
31
+ rename_args(self.mlx_language_model_handler_kwargs, "mlx_lm")
32
+ rename_args(self.parler_tts_handler_kwargs, "tts")
33
+ rename_args(self.melo_tts_handler_kwargs, "melo")
34
+ rename_args(self.chat_tts_handler_kwargs, "chat_tts")
35
 
36
  self.queues_and_events = initialize_queues_and_events()
37