Merge pull request #135 from jhj0517/fix/refactor

modules/faster_whisper_inference.py

@@ -246,7 +246,7 @@ class FasterWhisperInference(BaseInterface):
         """
         start_time = time.time()
 
-        params = WhisperGradioComponents.to_values(*whisper_params)
+        params = WhisperValues(*whisper_params)
 
         if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
             self.update_model(params.model_size, params.compute_type, progress)

modules/whisper_Inference.py

@@ -236,7 +236,7 @@ class WhisperInference(BaseInterface):
             elapsed time for transcription
         """
         start_time = time.time()
-        params = WhisperGradioComponents.to_values(*whisper_params)
+        params = WhisperValues(*whisper_params)
 
         if params.model_size != self.current_model_size or self.model is None or self.current_compute_type != params.compute_type:
             self.update_model(params.model_size, params.compute_type, progress)

modules/whisper_data_class.py

@@ -45,7 +45,8 @@ class WhisperGradioComponents:
 
     def to_list(self) -> list:
         """
-        Converts the data class attributes into a list, to pass parameters to a function before Gradio pre-processing.
+        Converts the data class attributes into a list, to pass parameters to a
+        button click event function before Gradio pre-processing.
 
         Returns
         ----------
@@ -53,23 +54,6 @@ class WhisperGradioComponents:
         """
         return [getattr(self, f.name) for f in fields(self)]
 
-    @staticmethod
-    def to_values(*params):
-        """
-        Convert a tuple of parameters into a WhisperValues data class, to use parameters in a function after Gradio pre-processing.
-
-        Parameters
-        ----------
-        *params: tuple
-            This is provided in a tuple because Gradio does not support **kwargs arbitrary.
-            Reference : https://discuss.huggingface.co/t/passing-an-additional-argument-to-a-function/25140/2
-
-        Returns
-        ----------
-        A WhisperValues data class
-        """
-        return WhisperValues(*params)
-
 
 @dataclass
 class WhisperValues:
@@ -83,6 +67,6 @@ class WhisperValues:
     best_of: int
     patience: float
     """
-    A data class to use Whisper parameters in the function after Gradio pre-processing.
+    A data class to use Whisper parameters in your function after Gradio pre-processing.
     See this documentation for more information about Gradio pre-processing: : https://www.gradio.app/docs/components
     """