configure device type based on availability

app.py

@@ -18,8 +18,16 @@ import contextlib
 from sklearn.cluster import AgglomerativeClustering
 import numpy as np
 
+if torch.cuda.is_available():
+    device_type = "cuda"
+elif torch.backends.mps.is_available():
+    device_type = "mps"
+else:
+    device_type = "cpu"
+
+
 embedding_model = PretrainedSpeakerEmbedding(
-    "speechbrain/spkrec-ecapa-voxceleb", device=torch.device("mps")
+    "speechbrain/spkrec-ecapa-voxceleb", device=torch.device(device_type)
 )
 
 audio = Audio()