LAP-DEV commited on
Commit
bbc6333
·
verified ·
1 Parent(s): a576509

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +2 -2
modules/whisper/whisper_base.py CHANGED
@@ -249,7 +249,7 @@ 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))
@@ -280,7 +280,7 @@ class WhisperBase(ABC):
280
  total_result += f'{info["subtitle"]}'
281
  total_time += info["time_for_task"]
282
  #total_info += f'{info["lang"]}'
283
- total_info += f"Input file: {info['input']}\nLanguage prediction: {info['lang']}"
284
 
285
  #result_str = f"Processing of file '{file_name}{file_ext}' done in {self.format_time(total_time)}:\n\n{total_result}"
286
  total_info += f"\nTranscription duration: {self.format_time(total_time)}"
 
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))
 
280
  total_result += f'{info["subtitle"]}'
281
  total_time += info["time_for_task"]
282
  #total_info += f'{info["lang"]}'
283
+ total_info += f"Input file: {info['input']}\nLanguage prediction: {info['lang']}\n"
284
 
285
  #result_str = f"Processing of file '{file_name}{file_ext}' done in {self.format_time(total_time)}:\n\n{total_result}"
286
  total_info += f"\nTranscription duration: {self.format_time(total_time)}"