diff --git a/audio.py b/audio.py
index 32b20c449..32ab5fabe 100644
--- a/audio.py
+++ b/audio.py
@@ -97,7 +97,7 @@ def _linear_to_mel(spectogram):
 
 def _build_mel_basis():
     assert hp.fmax <= hp.sample_rate // 2
-    return librosa.filters.mel(hp.sample_rate, hp.n_fft, n_mels=hp.num_mels,
+    return librosa.filters.mel(sr=hp.sample_rate, n_fft=hp.n_fft, n_mels=hp.num_mels,
                                fmin=hp.fmin, fmax=hp.fmax)
 
 def _amp_to_db(x):
diff --git a/face_detection/api.py b/face_detection/api.py
index cb02d5252..80ffa7258 100644
--- a/face_detection/api.py
+++ b/face_detection/api.py
@@ -46,7 +46,7 @@ def __int__(self):
 class FaceAlignment:
     def __init__(self, landmarks_type, network_size=NetworkSize.LARGE,
                  device='cuda', flip_input=False, face_detector='sfd', verbose=False):
-        self.device = device
+        self.device = 'cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu'
         self.flip_input = flip_input
         self.landmarks_type = landmarks_type
         self.verbose = verbose
@@ -76,4 +76,4 @@ def get_detections_for_batch(self, images):
             x1, y1, x2, y2 = map(int, d[:-1])
             results.append((x1, y1, x2, y2))
 
-        return results
\ No newline at end of file
+        return results
diff --git a/face_detection/detection/core.py b/face_detection/detection/core.py
index 0f8275e8e..a06a167a0 100644
--- a/face_detection/detection/core.py
+++ b/face_detection/detection/core.py
@@ -24,7 +24,7 @@ def __init__(self, device, verbose):
                 logger = logging.getLogger(__name__)
                 logger.warning("Detection running on CPU, this may be potentially slow.")
 
-        if 'cpu' not in device and 'cuda' not in device:
+        if 'cpu' not in device and 'cuda' not in device and 'mps' not in device:
             if verbose:
                 logger.error("Expected values for device are: {cpu, cuda} but got: %s", device)
             raise ValueError
diff --git a/inference.py b/inference.py
index 90692521e..ee111b05d 100644
--- a/inference.py
+++ b/inference.py
@@ -154,7 +154,7 @@ def datagen(frames, mels):
 		yield img_batch, mel_batch, frame_batch, coords_batch
 
 mel_step_size = 16
-device = 'cuda' if torch.cuda.is_available() else 'cpu'
+device = 'cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu'
 print('Using {} for inference.'.format(device))
 
 def _load(checkpoint_path):
diff --git a/preprocess.py b/preprocess.py
index 5322012ac..1c3d44eb8 100644
--- a/preprocess.py
+++ b/preprocess.py
@@ -59,9 +59,9 @@ def process_video_file(vfile, args, gpu_id):
 		preds = fa[gpu_id].get_detections_for_batch(np.asarray(fb))
 
 		for j, f in enumerate(preds):
-			i += 1
 			if f is None:
 				continue
+			i += 1
 
 			x1, y1, x2, y2 = f
 			cv2.imwrite(path.join(fulldir, '{}.jpg'.format(i)), fb[j][y1:y2, x1:x2])
@@ -110,4 +110,4 @@ def main(args):
 			continue
 
 if __name__ == '__main__':
-	main(args)
\ No newline at end of file
+	main(args)
diff --git a/requirements.txt b/requirements.txt
index bfd428ab9..ae9f627d0 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,8 +1,8 @@
-librosa==0.7.0
-numpy==1.17.1
-opencv-contrib-python>=4.2.0.34
-opencv-python==4.1.0.25
-torch==1.1.0
-torchvision==0.3.0
-tqdm==4.45.0
-numba==0.48
+librosa
+numpy
+opencv-contrib-python
+opencv-python
+torch
+torchvision
+tqdm
+numba