daihui.zhang
commited on
Commit
·
9494251
1
Parent(s):
0a6c788
rename filenames
Browse files
main.py
CHANGED
@@ -3,7 +3,7 @@ from transcribe.serve import WhisperTranscriptionService
|
|
3 |
from uuid import uuid1
|
4 |
from logging import getLogger
|
5 |
import numpy as np
|
6 |
-
from transcribe.
|
7 |
from contextlib import asynccontextmanager
|
8 |
from multiprocessing import Process, freeze_support
|
9 |
from fastapi.staticfiles import StaticFiles
|
@@ -33,7 +33,7 @@ async def get_audio_from_websocket(websocket)->np.array:
|
|
33 |
@asynccontextmanager
|
34 |
async def lifespan(app:FastAPI):
|
35 |
global pipe
|
36 |
-
pipe =
|
37 |
pipe.wait_ready()
|
38 |
logger.info("Pipeline is ready.")
|
39 |
yield
|
|
|
3 |
from uuid import uuid1
|
4 |
from logging import getLogger
|
5 |
import numpy as np
|
6 |
+
from transcribe.process_pipeline import ProcessingPipes
|
7 |
from contextlib import asynccontextmanager
|
8 |
from multiprocessing import Process, freeze_support
|
9 |
from fastapi.staticfiles import StaticFiles
|
|
|
33 |
@asynccontextmanager
|
34 |
async def lifespan(app:FastAPI):
|
35 |
global pipe
|
36 |
+
pipe = ProcessingPipes()
|
37 |
pipe.wait_ready()
|
38 |
logger.info("Pipeline is ready.")
|
39 |
yield
|
transcribe/helpers/vadprocessor.py
CHANGED
@@ -1,5 +1,4 @@
|
|
1 |
from copy import deepcopy
|
2 |
-
from queue import Queue, Empty
|
3 |
from time import time
|
4 |
from config import VAD_MODEL_PATH
|
5 |
from silero_vad import load_silero_vad
|
|
|
1 |
from copy import deepcopy
|
|
|
2 |
from time import time
|
3 |
from config import VAD_MODEL_PATH
|
4 |
from silero_vad import load_silero_vad
|
transcribe/{translatepipes.py → process_pipeline.py}
RENAMED
@@ -1,7 +1,7 @@
|
|
1 |
from .pipelines import WhisperPipe, MetaItem, WhisperChinese, Translate7BPipe, FunASRPipe, VadPipe, TranslatePipe
|
2 |
from .utils import timer
|
3 |
|
4 |
-
class
|
5 |
def __init__(self) -> None:
|
6 |
|
7 |
self._process = []
|
|
|
1 |
from .pipelines import WhisperPipe, MetaItem, WhisperChinese, Translate7BPipe, FunASRPipe, VadPipe, TranslatePipe
|
2 |
from .utils import timer
|
3 |
|
4 |
+
class ProcessingPipes:
|
5 |
def __init__(self) -> None:
|
6 |
|
7 |
self._process = []
|
transcribe/serve.py
CHANGED
@@ -10,9 +10,8 @@ import collections
|
|
10 |
from api_model import TransResult, Message
|
11 |
|
12 |
from .utils import log_block, start_thread, get_text_separator, filter_words
|
13 |
-
from .
|
14 |
-
|
15 |
-
from transcribe.pipelines import MetaItem
|
16 |
|
17 |
|
18 |
logger = getLogger("TranscriptionService")
|
@@ -25,7 +24,7 @@ class WhisperTranscriptionService:
|
|
25 |
"""
|
26 |
|
27 |
|
28 |
-
def __init__(self, websocket, pipe:
|
29 |
print('>>>>>>>>>>>>>>>> init service >>>>>>>>>>>>>>>>>>>>>>')
|
30 |
print('src_lang:', language)
|
31 |
self.source_language = language # 源语言
|
|
|
10 |
from api_model import TransResult, Message
|
11 |
|
12 |
from .utils import log_block, start_thread, get_text_separator, filter_words
|
13 |
+
from .process_pipeline import ProcessingPipes
|
14 |
+
from .pipelines import MetaItem
|
|
|
15 |
|
16 |
|
17 |
logger = getLogger("TranscriptionService")
|
|
|
24 |
"""
|
25 |
|
26 |
|
27 |
+
def __init__(self, websocket, pipe: ProcessingPipes, language=None, dst_lang=None, client_uid=None):
|
28 |
print('>>>>>>>>>>>>>>>> init service >>>>>>>>>>>>>>>>>>>>>>')
|
29 |
print('src_lang:', language)
|
30 |
self.source_language = language # 源语言
|