jhj0517 commited on
Commit
927f1ef
·
1 Parent(s): 5a3afa2

Rename model

Browse files
app.py CHANGED
@@ -196,7 +196,7 @@ class App:
196
  dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
197
 
198
  return (
199
- WhisperGradioComponents(
200
  model_size=dd_model, lang=dd_lang, is_translate=cb_translate, beam_size=nb_beam_size,
201
  log_prob_threshold=nb_log_prob_threshold, no_speech_threshold=nb_no_speech_threshold,
202
  compute_type=dd_compute_type, best_of=nb_best_of, patience=nb_patience,
 
196
  dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
197
 
198
  return (
199
+ TranscriptionPipelineGradioComponents(
200
  model_size=dd_model, lang=dd_lang, is_translate=cb_translate, beam_size=nb_beam_size,
201
  log_prob_threshold=nb_log_prob_threshold, no_speech_threshold=nb_no_speech_threshold,
202
  compute_type=dd_compute_type, best_of=nb_best_of, patience=nb_patience,
modules/whisper/faster_whisper_inference.py CHANGED
@@ -62,7 +62,7 @@ class FasterWhisperInference(WhisperBase):
62
  """
63
  start_time = time.time()
64
 
65
- params = WhisperGradioComponents.as_value(*whisper_params)
66
 
67
  if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
68
  self.update_model(params.model_size, params.compute_type, progress)
 
62
  """
63
  start_time = time.time()
64
 
65
+ params = TranscriptionPipelineGradioComponents.as_value(*whisper_params)
66
 
67
  if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
68
  self.update_model(params.model_size, params.compute_type, progress)
modules/whisper/insanely_fast_whisper_inference.py CHANGED
@@ -61,7 +61,7 @@ class InsanelyFastWhisperInference(WhisperBase):
61
  elapsed time for transcription
62
  """
63
  start_time = time.time()
64
- params = WhisperGradioComponents.as_value(*whisper_params)
65
 
66
  if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
67
  self.update_model(params.model_size, params.compute_type, progress)
 
61
  elapsed time for transcription
62
  """
63
  start_time = time.time()
64
+ params = TranscriptionPipelineGradioComponents.as_value(*whisper_params)
65
 
66
  if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
67
  self.update_model(params.model_size, params.compute_type, progress)
modules/whisper/whisper_Inference.py CHANGED
@@ -51,7 +51,7 @@ class WhisperInference(WhisperBase):
51
  elapsed time for transcription
52
  """
53
  start_time = time.time()
54
- params = WhisperGradioComponents.as_value(*whisper_params)
55
 
56
  if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
57
  self.update_model(params.model_size, params.compute_type, progress)
 
51
  elapsed time for transcription
52
  """
53
  start_time = time.time()
54
+ params = TranscriptionPipelineGradioComponents.as_value(*whisper_params)
55
 
56
  if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
57
  self.update_model(params.model_size, params.compute_type, progress)
modules/whisper/whisper_base.py CHANGED
@@ -99,7 +99,7 @@ class WhisperBase(ABC):
99
  elapsed_time: float
100
  elapsed time for running
101
  """
102
- params = WhisperGradioComponents.as_value(*whisper_params)
103
 
104
  self.cache_parameters(
105
  whisper_params=params,
 
99
  elapsed_time: float
100
  elapsed time for running
101
  """
102
+ params = TranscriptionPipelineGradioComponents.as_value(*whisper_params)
103
 
104
  self.cache_parameters(
105
  whisper_params=params,