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.translatepipes import TranslatePipes
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 = TranslatePipes()
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 TranslatePipes:
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 .translatepipes import TranslatePipes
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: TranslatePipes, language=None, dst_lang=None, client_uid=None):
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 # 源语言