jhj0517 commited on
Commit
e0e906d
·
1 Parent(s): 6c42d0c

Fix return type

Browse files
modules/whisper/base_transcription_pipeline.py CHANGED
@@ -73,7 +73,7 @@ class BaseTranscriptionPipeline(ABC):
73
  progress: gr.Progress = gr.Progress(),
74
  add_timestamp: bool = True,
75
  *pipeline_params,
76
- ) -> Tuple[List[dict], float]:
77
  """
78
  Run transcription with conditional pre-processing and post-processing.
79
  The VAD will be performed to remove noise from the audio input in pre-processing, if enabled.
@@ -93,8 +93,8 @@ class BaseTranscriptionPipeline(ABC):
93
 
94
  Returns
95
  ----------
96
- segments_result: List[dict]
97
- list of dicts that includes start, end timestamps and transcribed text
98
  elapsed_time: float
99
  elapsed time for running
100
  """
 
73
  progress: gr.Progress = gr.Progress(),
74
  add_timestamp: bool = True,
75
  *pipeline_params,
76
+ ) -> Tuple[List[Segment], float]:
77
  """
78
  Run transcription with conditional pre-processing and post-processing.
79
  The VAD will be performed to remove noise from the audio input in pre-processing, if enabled.
 
93
 
94
  Returns
95
  ----------
96
+ segments_result: List[Segment]
97
+ list of Segment that includes start, end timestamps and transcribed text
98
  elapsed_time: float
99
  elapsed time for running
100
  """