jhj0517 commited on
Commit
37be773
·
1 Parent(s): f137136

Fix to_list error

Browse files
Files changed (1) hide show
  1. modules/whisper/data_classes.py +7 -6
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 = [value for key, value in self.whisper.to_dict().items()]
492
- vad_list = [value for key, value in self.vad.to_dict().items()]
493
- diarization_list = [value for key, value in self.vad.to_dict().items()]
494
- bgm_sep_list = [value for key, value in self.bgm_separation.to_dict().items()]
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
- whisper_list, data_list = data_list[0:len(WhisperParams.__annotations__)]
 
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