will print transcribed text from long audios

long_audio_transcribe.py

@@ -27,6 +27,7 @@ if __name__ == "__main__":
         lang2token = {
             'zh': "[ZH]",
         }
+    assert(torch.cuda.is_available()), "Please enable GPU in order to run Whisper!"
     model = whisper.load_model(args.whisper_size)
     speaker_annos = []
     for file in filelist:

short_audio_transcribe.py

@@ -2,6 +2,7 @@ import whisper
 import os
 import torchaudio
 import argparse
+import torch
 
 lang2token = {
             'zh': "[ZH]",
@@ -47,6 +48,7 @@ if __name__ == "__main__":
         lang2token = {
             'zh': "[ZH]",
         }
+    assert (torch.cuda.is_available()), "Please enable GPU in order to run Whisper!"
     model = whisper.load_model(args.whisper_size)
     parent_dir = "./custom_character_voice/"
     speaker_names = list(os.walk(parent_dir))[0][1]