Update demo.py

demo.py

@@ -28,6 +28,7 @@ from utils import rotation_conversions as rc
 import soundfile as sf
 import librosa 
 import subprocess
+import shutil
 from transformers import pipeline
 from models.vq.model import RVQVAE
 
@@ -72,11 +73,13 @@ class BaseTrainer(object):
                 file.write(text)
             
             # use montreal forced aligner to get textgrid
-            # Run MFA with full conda environment PATH
-            conda_bin = "/Users/tharunsaireddy/miniforge3/envs/gesturelsm/bin"
+            mfa_override = os.environ.get("MFA_BINARY")
+            mfa_path = mfa_override or shutil.which("mfa")
+            if not mfa_path:
+                raise FileNotFoundError(
+                    "Montreal Forced Aligner binary not found. Install it or set MFA_BINARY"
+                )
             env = os.environ.copy()
-            env["PATH"] = f"{conda_bin}:{env.get('PATH', '')}"
-            mfa_path = f"{conda_bin}/mfa"
             command = [mfa_path, "align", tmp_dir, "english_us_arpa", "english_us_arpa", tmp_dir]
             result = subprocess.run(command, capture_output=True, text=True, env=env)
             print(f"MFA result: {result}")