jhj0517 commited on
Commit
732a962
·
1 Parent(s): 4495187

Add defaults to the function

Browse files
modules/whisper/insanely_fast_whisper_inference.py CHANGED
@@ -39,7 +39,7 @@ class InsanelyFastWhisperInference(WhisperBase):
39
 
40
  def transcribe(self,
41
  audio: Union[str, np.ndarray, torch.Tensor],
42
- progress: gr.Progress,
43
  *whisper_params,
44
  ) -> Tuple[List[dict], float]:
45
  """
@@ -98,7 +98,7 @@ class InsanelyFastWhisperInference(WhisperBase):
98
  def update_model(self,
99
  model_size: str,
100
  compute_type: str,
101
- progress: gr.Progress,
102
  ):
103
  """
104
  Update current model setting
 
39
 
40
  def transcribe(self,
41
  audio: Union[str, np.ndarray, torch.Tensor],
42
+ progress: gr.Progress = gr.Progress(),
43
  *whisper_params,
44
  ) -> Tuple[List[dict], float]:
45
  """
 
98
  def update_model(self,
99
  model_size: str,
100
  compute_type: str,
101
+ progress: gr.Progress = gr.Progress(),
102
  ):
103
  """
104
  Update current model setting
modules/whisper/whisper_Inference.py CHANGED
@@ -28,7 +28,7 @@ class WhisperInference(WhisperBase):
28
 
29
  def transcribe(self,
30
  audio: Union[str, np.ndarray, torch.Tensor],
31
- progress: gr.Progress,
32
  *whisper_params,
33
  ) -> Tuple[List[dict], float]:
34
  """
@@ -79,7 +79,7 @@ class WhisperInference(WhisperBase):
79
  def update_model(self,
80
  model_size: str,
81
  compute_type: str,
82
- progress: gr.Progress,
83
  ):
84
  """
85
  Update current model setting
 
28
 
29
  def transcribe(self,
30
  audio: Union[str, np.ndarray, torch.Tensor],
31
+ progress: gr.Progress = gr.Progress(),
32
  *whisper_params,
33
  ) -> Tuple[List[dict], float]:
34
  """
 
79
  def update_model(self,
80
  model_size: str,
81
  compute_type: str,
82
+ progress: gr.Progress = gr.Progress(),
83
  ):
84
  """
85
  Update current model setting