Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -234,9 +234,10 @@ class WhisperBase(ABC):
         return result, elapsed_time
 
     def transcribe_file(self,
-                        files: Optional[List] = None,
+                        files_audio: Optional[List] = None,
+                        files_video: Optional[List] = None,
                         files_multi: Optional[List] = None,
-                        input_multi: str = "Single",
+                        input_multi: str = "Audio",
                         input_folder_path: Optional[str] = None,
                         file_format: list = ["CSV"],
                         add_timestamp: bool = True,
@@ -252,8 +253,10 @@ class WhisperBase(ABC):
 
         Parameters
         ----------
-        files: list
-            List of files to transcribe from gr.Files()
+        files_audio: list
+            List of files to transcribe from gr.Audio()
+        files_video: list
+            List of files to transcribe from gr.Video()
         files_multi: list
             List of files to transcribe from gr.Files_multi()
         input_multi: bool
@@ -288,7 +291,13 @@ class WhisperBase(ABC):
 
         try:
             file_count_total = 0
-            if input_multi != "Single":
+            files = ""
+
+            if input_multi == "Audio":
+                files = files_audio
+            elif input_multi == "Video":
+                files = files_video
+            else:
                 files = files_multi
                 file_count_total = len(files)