jhj0517 commited on
Commit
6148cfe
1 Parent(s): ca8ee6a

rename function

Browse files
app.py CHANGED
@@ -149,7 +149,7 @@ class App:
149
  hf_token=tb_hf_token)
150
 
151
  btn_run.click(fn=self.whisper_inf.transcribe_file,
152
- inputs=params + whisper_params.to_list(),
153
  outputs=[tb_indicator, files_subtitles])
154
  btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
155
  dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
@@ -231,7 +231,7 @@ class App:
231
  hf_token=tb_hf_token)
232
 
233
  btn_run.click(fn=self.whisper_inf.transcribe_youtube,
234
- inputs=params + whisper_params.to_list(),
235
  outputs=[tb_indicator, files_subtitles])
236
  tb_youtubelink.change(get_ytmetas, inputs=[tb_youtubelink],
237
  outputs=[img_thumbnail, tb_title, tb_description])
@@ -305,7 +305,7 @@ class App:
305
  hf_token=tb_hf_token)
306
 
307
  btn_run.click(fn=self.whisper_inf.transcribe_mic,
308
- inputs=params + whisper_params.to_list(),
309
  outputs=[tb_indicator, files_subtitles])
310
  btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
311
  dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
 
149
  hf_token=tb_hf_token)
150
 
151
  btn_run.click(fn=self.whisper_inf.transcribe_file,
152
+ inputs=params + whisper_params.as_list(),
153
  outputs=[tb_indicator, files_subtitles])
154
  btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
155
  dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
 
231
  hf_token=tb_hf_token)
232
 
233
  btn_run.click(fn=self.whisper_inf.transcribe_youtube,
234
+ inputs=params + whisper_params.as_list(),
235
  outputs=[tb_indicator, files_subtitles])
236
  tb_youtubelink.change(get_ytmetas, inputs=[tb_youtubelink],
237
  outputs=[img_thumbnail, tb_title, tb_description])
 
305
  hf_token=tb_hf_token)
306
 
307
  btn_run.click(fn=self.whisper_inf.transcribe_mic,
308
+ inputs=params + whisper_params.as_list(),
309
  outputs=[tb_indicator, files_subtitles])
310
  btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
311
  dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
modules/faster_whisper_inference.py CHANGED
@@ -52,7 +52,7 @@ class FasterWhisperInference(WhisperBase):
52
  """
53
  start_time = time.time()
54
 
55
- params = WhisperParameters.post_process(*whisper_params)
56
 
57
  if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
58
  self.update_model(params.model_size, params.compute_type, progress)
 
52
  """
53
  start_time = time.time()
54
 
55
+ params = WhisperParameters.as_value(*whisper_params)
56
 
57
  if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
58
  self.update_model(params.model_size, params.compute_type, progress)
modules/insanely_fast_whisper_inference.py CHANGED
@@ -53,7 +53,7 @@ class InsanelyFastWhisperInference(WhisperBase):
53
  elapsed time for transcription
54
  """
55
  start_time = time.time()
56
- params = WhisperParameters.post_process(*whisper_params)
57
 
58
  if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
59
  self.update_model(params.model_size, params.compute_type, progress)
 
53
  elapsed time for transcription
54
  """
55
  start_time = time.time()
56
+ params = WhisperParameters.as_value(*whisper_params)
57
 
58
  if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
59
  self.update_model(params.model_size, params.compute_type, progress)
modules/whisper_Inference.py CHANGED
@@ -45,7 +45,7 @@ class WhisperInference(WhisperBase):
45
  elapsed time for transcription
46
  """
47
  start_time = time.time()
48
- params = WhisperParameters.post_process(*whisper_params)
49
 
50
  if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
51
  self.update_model(params.model_size, params.compute_type, progress)
 
45
  elapsed time for transcription
46
  """
47
  start_time = time.time()
48
+ params = WhisperParameters.as_value(*whisper_params)
49
 
50
  if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
51
  self.update_model(params.model_size, params.compute_type, progress)
modules/whisper_parameter.py CHANGED
@@ -133,7 +133,7 @@ class WhisperParameters:
133
  Read more about : https://huggingface.co/pyannote/speaker-diarization-3.1#requirements
134
  """
135
 
136
- def to_list(self) -> list:
137
  """
138
  Converts the data class attributes into a list, Use in Gradio UI before Gradio pre-processing.
139
  See more about Gradio pre-processing: : https://www.gradio.app/docs/components
@@ -145,7 +145,7 @@ class WhisperParameters:
145
  return [getattr(self, f.name) for f in fields(self)]
146
 
147
  @staticmethod
148
- def post_process(*args) -> 'WhisperValues':
149
  """
150
  To use Whisper parameters in function after Gradio post-processing.
151
  See more about Gradio post-processing: : https://www.gradio.app/docs/components
 
133
  Read more about : https://huggingface.co/pyannote/speaker-diarization-3.1#requirements
134
  """
135
 
136
+ def as_list(self) -> list:
137
  """
138
  Converts the data class attributes into a list, Use in Gradio UI before Gradio pre-processing.
139
  See more about Gradio pre-processing: : https://www.gradio.app/docs/components
 
145
  return [getattr(self, f.name) for f in fields(self)]
146
 
147
  @staticmethod
148
+ def as_value(*args) -> 'WhisperValues':
149
  """
150
  To use Whisper parameters in function after Gradio post-processing.
151
  See more about Gradio post-processing: : https://www.gradio.app/docs/components