LAP-DEV commited on
Commit
7c29e28
·
verified ·
1 Parent(s): e7febdc

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +3 -2
modules/whisper/whisper_base.py CHANGED
@@ -626,8 +626,9 @@ class WhisperBase(ABC):
626
  temp_line_items = temp_line.split("#TAB#")
627
 
628
  # Add columns to match Dataframe
629
- if timestamps==False and diarize==True: temp_line_items.insert(0,"")
630
- if timestamps==True and diarize==False: temp_line_items.insert(1,"")
 
631
 
632
  for temp_line_item in temp_line_items:
633
  temp_line_list.append(temp_line_item)
 
626
  temp_line_items = temp_line.split("#TAB#")
627
 
628
  # Add columns to match Dataframe
629
+ if len(temp_line_items)<3:
630
+ if timestamps==False and diarize==True: temp_line_items.insert(0,"")
631
+ if timestamps==True and diarize==False: temp_line_items.insert(1,"")
632
 
633
  for temp_line_item in temp_line_items:
634
  temp_line_list.append(temp_line_item)