LAP-DEV commited on
Commit
12318c7
·
verified ·
1 Parent(s): 690ec2b

Update modules/utils/subtitle_manager.py

Browse files
Files changed (1) hide show
  1. modules/utils/subtitle_manager.py +5 -0
modules/utils/subtitle_manager.py CHANGED
@@ -51,6 +51,11 @@ def get_csv(segments):
51
 
52
  # Check if speakers are identified and get speaker id & text
53
  temp = re.search(r'SPEAKER_[0-9][0-9]: ',segment['text'])
 
 
 
 
 
54
  if temp != None:
55
  temp_string = str(temp.group())
56
  speaker_id = temp_string.replace(': ','')
 
51
 
52
  # Check if speakers are identified and get speaker id & text
53
  temp = re.search(r'SPEAKER_[0-9][0-9]: ',segment['text'])
54
+
55
+ # check if speaker 'None' is identified for e.g. music
56
+ if temp == None:
57
+ temp = re.search('None: ',segment['text'])
58
+
59
  if temp != None:
60
  temp_string = str(temp.group())
61
  speaker_id = temp_string.replace(': ','')