Enhance device selection logic in setup_embedding_model to support MPS for Apple Silicon, in addition to CUDA and CPU fallback.

data_setup.py

@@ -111,7 +111,12 @@ def setup_embedding_model(model_name):
         model_name (str): Name or path of the SentenceTransformer model
     """
     print(f"Setting up language model: {time.strftime('%Y-%m-%d %H:%M:%S')}")
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    if torch.cuda.is_available():
+        device = torch.device("cuda")
+    elif torch.backends.mps.is_available():
+        device = torch.device("mps")
+    else:
+        device = torch.device("cpu")
     print(f"Using device: {device}")
     
     model = SentenceTransformer(model_name)