Set output transcription file as srt into transcrie.py
Browse files- transcribe.py +2 -1
transcribe.py
CHANGED
|
@@ -13,9 +13,10 @@ def main(args):
|
|
| 13 |
device = "cuda"
|
| 14 |
verbose = False
|
| 15 |
threads = 4
|
|
|
|
| 16 |
command = f'whisper --model {model} --output_dir {output_folder} --language {language} \
|
| 17 |
--word_timestamps {word_timestamps} --fp16 {fp16} --device {device} --verbose {verbose} \
|
| 18 |
-
--threads {threads} {audio_file}'
|
| 19 |
os.system(command)
|
| 20 |
|
| 21 |
if __name__ == "__main__":
|
|
|
|
| 13 |
device = "cuda"
|
| 14 |
verbose = False
|
| 15 |
threads = 4
|
| 16 |
+
output_format = "srt"
|
| 17 |
command = f'whisper --model {model} --output_dir {output_folder} --language {language} \
|
| 18 |
--word_timestamps {word_timestamps} --fp16 {fp16} --device {device} --verbose {verbose} \
|
| 19 |
+
--threads {threads} --output_format {output_format} {audio_file}'
|
| 20 |
os.system(command)
|
| 21 |
|
| 22 |
if __name__ == "__main__":
|