LAP-DEV commited on
Commit
4d42d8e
·
verified ·
1 Parent(s): 1ae2f46

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +3 -3
modules/whisper/whisper_base.py CHANGED
@@ -431,9 +431,10 @@ class WhisperBase(ABC):
431
  total_info += f"\nProcessed {file_count} {temp_file_count_text} in {self.format_time((time_end-time_start).total_seconds())}"
432
 
433
  result_str = total_result.rstrip("\n")
 
434
  result_file_path = [info['path'] for info in files_to_download.values()]
435
-
436
- return [self.transform_text_to_list(result_str),result_file_path,total_info]
437
 
438
  except Exception as e:
439
  print(f"Error transcribing file: {e}")
@@ -626,7 +627,6 @@ class WhisperBase(ABC):
626
  gc.collect()
627
 
628
  def transform_text_to_list(inputdata):
629
- print("Transform: "+inputdata)
630
  outputdata = []
631
  temp_inputdata = (inputdata.strip("\n")).splitlines()
632
  for temp_line in temp_inputdata:
 
431
  total_info += f"\nProcessed {file_count} {temp_file_count_text} in {self.format_time((time_end-time_start).total_seconds())}"
432
 
433
  result_str = total_result.rstrip("\n")
434
+ result_str = self.transform_text_to_list(result_str)
435
  result_file_path = [info['path'] for info in files_to_download.values()]
436
+
437
+ return [result_str,result_file_path,total_info]
438
 
439
  except Exception as e:
440
  print(f"Error transcribing file: {e}")
 
627
  gc.collect()
628
 
629
  def transform_text_to_list(inputdata):
 
630
  outputdata = []
631
  temp_inputdata = (inputdata.strip("\n")).splitlines()
632
  for temp_line in temp_inputdata: