Spaces:
Running
Running
jhj0517
commited on
Commit
·
37be773
1
Parent(s):
f137136
Fix to_list error
Browse files
modules/whisper/data_classes.py
CHANGED
@@ -488,24 +488,25 @@ class TranscriptionPipelineParams(BaseModel):
|
|
488 |
Related Gradio issue: https://github.com/gradio-app/gradio/issues/2471
|
489 |
See more about Gradio pre-processing: https://www.gradio.app/docs/components
|
490 |
"""
|
491 |
-
whisper_list =
|
492 |
-
vad_list =
|
493 |
-
diarization_list =
|
494 |
-
bgm_sep_list =
|
495 |
return whisper_list + vad_list + diarization_list + bgm_sep_list
|
496 |
|
497 |
@staticmethod
|
498 |
def from_list(pipeline_list: List) -> 'TranscriptionPipelineParams':
|
499 |
"""Convert list to the data class again to use it in a function."""
|
500 |
data_list = deepcopy(pipeline_list)
|
501 |
-
|
|
|
502 |
data_list = data_list[len(WhisperParams.__annotations__):]
|
503 |
|
504 |
vad_list = data_list[0:len(VadParams.__annotations__)]
|
505 |
data_list = data_list[len(VadParams.__annotations__):]
|
506 |
|
507 |
diarization_list = data_list[0:len(DiarizationParams.__annotations__)]
|
508 |
-
data_list = data_list[len(DiarizationParams.__annotations__)]
|
509 |
|
510 |
bgm_sep_list = data_list[0:len(BGMSeparationParams.__annotations__)]
|
511 |
|
|
|
488 |
Related Gradio issue: https://github.com/gradio-app/gradio/issues/2471
|
489 |
See more about Gradio pre-processing: https://www.gradio.app/docs/components
|
490 |
"""
|
491 |
+
whisper_list = self.whisper.to_list()
|
492 |
+
vad_list = self.vad.to_list()
|
493 |
+
diarization_list = self.diarization.to_list()
|
494 |
+
bgm_sep_list = self.bgm_separation.to_list()
|
495 |
return whisper_list + vad_list + diarization_list + bgm_sep_list
|
496 |
|
497 |
@staticmethod
|
498 |
def from_list(pipeline_list: List) -> 'TranscriptionPipelineParams':
|
499 |
"""Convert list to the data class again to use it in a function."""
|
500 |
data_list = deepcopy(pipeline_list)
|
501 |
+
|
502 |
+
whisper_list = data_list[0:len(WhisperParams.__annotations__)]
|
503 |
data_list = data_list[len(WhisperParams.__annotations__):]
|
504 |
|
505 |
vad_list = data_list[0:len(VadParams.__annotations__)]
|
506 |
data_list = data_list[len(VadParams.__annotations__):]
|
507 |
|
508 |
diarization_list = data_list[0:len(DiarizationParams.__annotations__)]
|
509 |
+
data_list = data_list[len(DiarizationParams.__annotations__):]
|
510 |
|
511 |
bgm_sep_list = data_list[0:len(BGMSeparationParams.__annotations__)]
|
512 |
|