Update soni_translate/speech_segmentation.py

soni_translate/speech_segmentation.py

@@ -57,7 +57,7 @@ def load_align_and_align_segments(result, audio, DAMHF):
     # Load alignment model
     model_a, metadata = whisperx.load_align_model(
         language_code=result["language"],
-        device=os.environ.get("SONITR_DEVICE") if os.environ.get("ZERO_GPU") != "TRUE" else "cuda",
+        device=os.environ.get("SONITR_DEVICE") if os.environ.get("ZERO_GPU") != "TRUE" else "cpu",
         model_name=None
         if result["language"] in DAMHF.keys()
         else EXTRA_ALIGN[result["language"]],
@@ -69,7 +69,7 @@ def load_align_and_align_segments(result, audio, DAMHF):
         model_a,
         metadata,
         audio,
-        os.environ.get("SONITR_DEVICE") if os.environ.get("ZERO_GPU") != "TRUE" else "cuda",
+        os.environ.get("SONITR_DEVICE") if os.environ.get("ZERO_GPU") != "TRUE" else "cpu",
         return_char_alignments=True,
         print_progress=False,
     )