fix logging path; enable CUDA for whispher

Former-commit-id: 0ea328b0a70e95f6061b93083ed418eafa4857c8

pipeline.py

@@ -9,6 +9,7 @@ import whisper
 from srt2ass import srt2ass
 import logging
 from datetime import datetime
+import torch
 
 import subprocess
 
@@ -109,7 +110,10 @@ def get_srt_class(srt_file_en, result_path, video_name, audio_path, audio_file =
 
             # use stable-whisper
             elif method == "stable":
-                model = stable_whisper.load_model(whisper_model)
+
+                # use cuda if available
+                devices = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") 
+                model = stable_whisper.load_model(whisper_model, device = devices)
                 transcript = model.transcribe(audio_path, regroup = False, initial_prompt="Hello, welcome to my lecture. Are you good my friend?")
                 (
                     transcript
@@ -265,7 +269,9 @@ def main():
 
     audio_path, audio_file, video_path, VIDEO_NAME = get_sources(args, DOWNLOAD_PATH, RESULT_PATH, VIDEO_NAME)
 
-    logging.basicConfig(level=logging.INFO, handlers=[logging.FileHandler("{}/{}_{}.log".format(args.log_dir, VIDEO_NAME, datetime.now().strftime("%m%d%Y_%H%M%S")))], encoding='utf-8')
+    if not os.path.exists(args.log_dir):
+        os.makedirs(args.log_dir)
+    logging.basicConfig(level=logging.INFO, handlers=[logging.FileHandler("{}/{}_{}.log".format(args.log_dir, VIDEO_NAME, datetime.now().strftime("%m%d%Y_%H%M%S")), 'w', encoding='utf-8')])
     logging.info("---------------------Video Info---------------------")
     logging.info("Video name: {}, translation model: {}, video link: {}".format(VIDEO_NAME, args.model_name, args.link))