LAP-DEV commited on
Commit
98317b9
·
verified ·
1 Parent(s): 8267c37

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +3 -1
modules/whisper/whisper_base.py CHANGED
@@ -249,7 +249,9 @@ class WhisperBase(ABC):
249
  file_format=file_format,
250
  output_dir=self.output_dir
251
  )
252
- files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language, "input": (file_name+file_ext)}
 
 
253
 
254
  ## Add output file as txt
255
  file_name, file_ext = os.path.splitext(os.path.basename(file))
 
249
  file_format=file_format,
250
  output_dir=self.output_dir
251
  )
252
+ #files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language, "input": (file_name+file_ext)}
253
+ input_file=str((file_name+file_ext))
254
+ files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language, "input": input_file}
255
 
256
  ## Add output file as txt
257
  file_name, file_ext = os.path.splitext(os.path.basename(file))