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,
|
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 |
-
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
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 |
|