jhj0517 commited on
Commit
6bc716a
·
1 Parent(s): 3b69814

Refactor parameter

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +4 -4
modules/whisper/whisper_base.py CHANGED
@@ -212,19 +212,19 @@ class WhisperBase(ABC):
212
  files = get_media_files(input_folder_path)
213
  if isinstance(files, str):
214
  files = [files]
215
- if files and not isinstance(files[0], gr.utils.NamedString):
216
- files = format_gradio_files(files)
217
 
218
  files_info = {}
219
  for file in files:
220
  transcribed_segments, time_for_task = self.run(
221
- file.name,
222
  progress,
223
  add_timestamp,
224
  *whisper_params,
225
  )
226
 
227
- file_name, file_ext = os.path.splitext(os.path.basename(file.name))
228
  subtitle, file_path = self.generate_and_write_file(
229
  file_name=file_name,
230
  transcribed_segments=transcribed_segments,
 
212
  files = get_media_files(input_folder_path)
213
  if isinstance(files, str):
214
  files = [files]
215
+ if files and isinstance(files[0], gr.utils.NamedString):
216
+ files = [file.name for file in files]
217
 
218
  files_info = {}
219
  for file in files:
220
  transcribed_segments, time_for_task = self.run(
221
+ file,
222
  progress,
223
  add_timestamp,
224
  *whisper_params,
225
  )
226
 
227
+ file_name, file_ext = os.path.splitext(os.path.basename(file))
228
  subtitle, file_path = self.generate_and_write_file(
229
  file_name=file_name,
230
  transcribed_segments=transcribed_segments,