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 # 源语言
|