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 |
|