solve: empty time stamp

Former-commit-id: 8f4b494099485ddd0256fbcf7eb99d537826d6ab

SRT.py

@@ -1,6 +1,4 @@
 from datetime import timedelta
-import os
-import whisper
 from csv import reader
 import re
 
@@ -10,6 +8,10 @@ class SRT_segment(object):
             segment = args[0]
             start_ms = int((segment['start']*100)%100*10)
             end_ms = int((segment['end']*100)%100*10)
+
+            if start_ms == end_ms and int(segment['start']) == int(segment['end']): # avoid empty time stamp
+                end_ms+=500
+
             start_time = str(timedelta(seconds=int(segment['start']), milliseconds=start_ms))
             end_time = str(timedelta(seconds=int(segment['end']), milliseconds=end_ms))
             if start_ms == 0:

pipeline.py

@@ -2,7 +2,6 @@ import openai
 from pytube import YouTube
 import argparse
 import os
-import whisper
 from tqdm import tqdm
 from SRT import SRT_script
 import stable_whisper