Merge pull request #6

Merged pull request #6

src/_LocalTranscribe.py

@@ -2,7 +2,7 @@ import os
 import datetime
 from glob import glob
 import whisper
-from torch import cuda, Generator
+from torch import backends, cuda, Generator
 import colorama
 from colorama import Back,Fore
 colorama.init(autoreset=True)
@@ -41,10 +41,13 @@ def transcribe(path, glob_file, model=None, language=None, verbose=False):
 
     """
     # Check for GPU acceleration
-    if cuda.is_available():
+    if backends.mps.is_available():
+        Generator('mps').manual_seed(42)
+    elif cuda.is_available():
         Generator('cuda').manual_seed(42)
     else:
         Generator().manual_seed(42)
+
     # Load model
     model = whisper.load_model(model)
     # Start main loop