diff --git a/src/_LocalTranscribe.py b/src/_LocalTranscribe.py
index 5d78562..adf1025 100644
--- a/src/_LocalTranscribe.py
+++ b/src/_LocalTranscribe.py
@@ -40,16 +40,19 @@ def transcribe(path, glob_file, model=None, language=None, verbose=False):
           within the specified path.
 
     """
-    # Check for GPU acceleration
+    # Check for GPU acceleration and set device
     if backends.mps.is_available():
+        device = 'mps'
         Generator('mps').manual_seed(42)
     elif cuda.is_available():
+        device = 'cuda'
         Generator('cuda').manual_seed(42)
     else:
+        device = 'cpu'
         Generator().manual_seed(42)
 
-    # Load model
-    model = whisper.load_model(model)
+    # Load model on the correct device
+    model = whisper.load_model(model, device=device)
     # Start main loop
     files_transcripted=[]   
     for file in glob_file:
diff --git a/src/__pycache__/_LocalTranscribe.cpython-312.pyc b/src/__pycache__/_LocalTranscribe.cpython-312.pyc
new file mode 100644
index 0000000..193465a
Binary files /dev/null and b/src/__pycache__/_LocalTranscribe.cpython-312.pyc differ