Update requirements.txt

requirements.txt

@@ -1,23 +1,25 @@
-# JAX stack (phải cố định version cũ, tương thích với Haiku)
-jax==0.4.13
-jaxlib==0.4.13
+# JAX stack (tương thích với haiku)
+jax==0.4.26
+jaxlib==0.4.26
 dm-haiku==0.0.11
 optax==0.1.7
 
-# Torch + Transformers
+# Audio + khoa học
+numpy==1.26.4
+scipy==1.11.3
+soundfile==0.12.1
+librosa==0.10.2.post1
+
+# Torch (CPU, Hugging Face Spaces chỉ hỗ trợ CPU nếu không bật GPU)
 torch==2.2.2
 torchaudio==2.2.2
+
+# Transformers (nếu vietTTS cần)
 transformers==4.41.2
 sentencepiece==0.2.0
 protobuf==4.25.3
 
-# Audio processing
-numpy==1.26.4
-scipy==1.11.3
-soundfile==0.12.1
-librosa==0.10.2.post1
-
-# Web & API
+# Gradio + web
 gradio==4.44.0
 fastapi==0.110.0
 uvicorn==0.30.1