Maximofn commited on
Commit
7bfb36e
1 Parent(s): 8dd5680

Set output transcription file as srt into transcrie.py

Browse files
Files changed (1) hide show
  1. 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__":