Skip to content

Commit 9c1fcdb

Browse files
authored
added end of the text marker
1 parent c92a373 commit 9c1fcdb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

caption_process.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def main():
4747
continue
4848
if speaker_stephen(text):
4949
stephen_speaking = True
50-
text = "".join(x[1] for x in re.findall(r">> (Stephen:|STEPHEN:|stephen:) (.*)$", text))
50+
text = "".join(x[1] for x in re.findall(r"&gt;&gt; (Stephen:|STEPHEN:|stephen:) (.*)$", text+<"|endoftext|">))
5151
# print(text)
5252
if not stephen_speaking:
5353
continue

0 commit comments

Comments
 (0)