LAP-DEV commited on
Commit
fae5dd9
·
verified ·
1 Parent(s): 798dc27

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +4 -4
modules/whisper/whisper_base.py CHANGED
@@ -622,15 +622,15 @@ class WhisperBase(ABC):
622
  outputdata = []
623
  temp_inputdata = inputdata.split("#NEWLINE#")
624
  for temp_line in temp_inputdata:
625
- temp_line_list = []
626
- temp_line_items = temp_line.split("#TAB#")
627
 
628
  # Add columns to match Dataframe
629
  if timestamps==True and diarize==False:
630
- temp_line_items.insert(1,"")
631
  if timestamps==False and diarize==True:
632
- temp_line_items.insert(0,"")
633
 
 
634
  for temp_line_item in temp_line_items:
635
  temp_line_list.append(temp_line_item)
636
  outputdata.append(temp_line_list)
 
622
  outputdata = []
623
  temp_inputdata = inputdata.split("#NEWLINE#")
624
  for temp_line in temp_inputdata:
625
+ temp_line_list = []
 
626
 
627
  # Add columns to match Dataframe
628
  if timestamps==True and diarize==False:
629
+ temp_line_items.insert(1,"#TAB#")
630
  if timestamps==False and diarize==True:
631
+ temp_line_items.insert(0,"#TAB#")
632
 
633
+ temp_line_items = temp_line.split("#TAB#")
634
  for temp_line_item in temp_line_items:
635
  temp_line_list.append(temp_line_item)
636
  outputdata.append(temp_line_list)