Update requirements.txt

requirements.txt

@@ -1,289 +1,32 @@
-# Web framework
-flask>=2.3.0,<4.0.0
+# --- Web / UI ---
+gradio[oauth]==4.44.1
+fastapi>=0.111,<0.114
+uvicorn[standard]>=0.20,<0.31
 
-# Core ML/AI packages
-torch>=2.0.0,<3.0.0
-transformers>=4.30.0,<5.0.0
-tokenizers>=0.13.0,<1.0.0
-accelerate>=0.20.0,<1.0.0
-safetensors>=0.3.0,<1.0.0
+# --- Core ML ---
+torch>=2.0,<3.0
+transformers>=4.45,<5.0
+tokenizers>=0.15,<0.22
+accelerate>=0.30,<0.35
+safetensors>=0.4,<1.0
+huggingface_hub>=0.20,<1.0
 
-# Hugging Face
-huggingface_hub>=0.20.0,<1.0.0
+# --- Data / utils ---
+numpy>=1.23,<2.0
+pandas>=1.5,<3.0
+scipy>=1.10,<2.0
+scikit-learn>=1.3,<2.0
 
-# Data processing
-numpy>=1.21.0,<2.0.0
-pandas>=1.3.0,<3.0.0
-scipy>=1.7.0,<2.0.0
-scikit-learn>=1.0.0,<2.0.0
+# --- Audio / media ---
+soundfile>=0.12,<1.0
+librosa>=0.10,<1.0
+pydub>=0.25
+ffmpeg-python>=0.2,<1.0
 
-# Audio processing
-soundfile>=0.12.0,<1.0.0
-librosa>=0.10.0,<1.0.0
-pydub>=0.25.0
-ffmpeg-python>=0.2.0,<1.0.0
-
-# Speech analysis - try specific versions that work better on CPU
-# This file was autogenerated by uv via the following command:
-#    uv pip compile ./setup.py
-accelerate==1.5.1
-    # via peft
-annotated-types==0.7.0
-    # via pydantic
-blobfile==3.0.0
-    # via batchalign (./setup.py)
-certifi==2025.1.31
-    # via requests
-cffi==1.17.1
-    # via soundfile
-charset-normalizer==3.4.1
-    # via requests
-click==8.1.8
-    # via
-    #   batchalign (./setup.py)
-    #   nltk
-    #   rich-click
-contourpy==1.3.1
-    # via matplotlib
-coverage==5.5
-    # via eyed3
-cycler==0.12.1
-    # via matplotlib
-deprecation==2.1.0
-    # via eyed3
-docopt==0.6.2
-    # via num2words
-emoji==2.14.1
-    # via stanza
-eyed3==0.9.7
-    # via batchalign (./setup.py)
-filelock==3.17.0
-    # via
-    #   blobfile
-    #   huggingface-hub
-    #   torch
-    #   transformers
-filetype==1.2.0
-    # via eyed3
-fonttools==4.56.0
-    # via matplotlib
-fsspec==2025.3.0
-    # via
-    #   huggingface-hub
-    #   torch
-hmmlearn==0.3.0
-    # via batchalign (./setup.py)
-huggingface-hub==0.29.3
-    # via
-    #   accelerate
-    #   peft
-    #   tokenizers
-    #   transformers
-idna==3.10
-    # via requests
-imbalanced-learn==0.13.0
-    # via imblearn
-imblearn==0.0
-    # via batchalign (./setup.py)
-jinja2==3.1.6
-    # via torch
-joblib==1.4.2
-    # via
-    #   imbalanced-learn
-    #   nltk
-    #   scikit-learn
-kiwisolver==1.4.8
-    # via matplotlib
-lxml==5.3.1
-    # via blobfile
-markdown-it-py==3.0.0
-    # via rich
-markupsafe==3.0.2
-    # via jinja2
-matplotlib==3.10.1
-    # via batchalign (./setup.py)
-mdurl==0.1.2
-    # via markdown-it-py
-mpmath==1.3.0
-    # via sympy
-narwhals==1.30.0
-    # via plotly
-networkx==3.4.2
-    # via
-    #   stanza
-    #   torch
-nltk==3.9.1
-    # via batchalign (./setup.py)
-num2words==0.5.14
-    # via batchalign (./setup.py)
-numpy==2.2.3
-    # via
-    #   accelerate
-    #   contourpy
-    #   hmmlearn
-    #   imbalanced-learn
-    #   matplotlib
-    #   peft
-    #   scikit-learn
-    #   scipy
-    #   stanza
-    #   transformers
-packaging==24.2
-    # via
-    #   accelerate
-    #   deprecation
-    #   huggingface-hub
-    #   matplotlib
-    #   peft
-    #   plotly
-    #   transformers
-peft==0.14.0
-    # via stanza
-pillow==11.1.0
-    # via matplotlib
-plotly==6.0.0
-    # via batchalign (./setup.py)
-praatio==6.0.1
-    # via batchalign (./setup.py)
-protobuf==6.30.0
-    # via stanza
-psutil==7.0.0
-    # via
-    #   accelerate
-    #   peft
-pycountry==24.6.1
-    # via batchalign (./setup.py)
-pycparser==2.22
-    # via cffi
-pycryptodomex==3.21.0
-    # via blobfile
-pydantic==2.10.6
-    # via batchalign (./setup.py)
-pydantic-core==2.27.2
-    # via pydantic
-pydub==0.25.1
-    # via batchalign (./setup.py)
-pyfiglet==1.0.2
-    # via batchalign (./setup.py)
-pygments==2.19.1
-    # via rich
-pyparsing==3.2.1
-    # via matplotlib
-python-dateutil==2.9.0.post0
-    # via matplotlib
-pyyaml==6.0.2
-    # via
-    #   accelerate
-    #   huggingface-hub
-    #   peft
-    #   transformers
-regex==2024.11.6
-    # via
-    #   nltk
-    #   tiktoken
-    #   transformers
-requests==2.32.3
-    # via
-    #   huggingface-hub
-    #   rev-ai
-    #   stanza
-    #   tiktoken
-    #   transformers
-rev-ai==2.21.0
-    # via batchalign (./setup.py)
-rich==13.9.4
-    # via
-    #   batchalign (./setup.py)
-    #   rich-click
-rich-click==1.8.8
-    # via batchalign (./setup.py)
-safetensors==0.5.3
-    # via
-    #   accelerate
-    #   peft
-    #   transformers
-scikit-learn==1.6.1
-    # via
-    #   hmmlearn
-    #   imbalanced-learn
-    #   sklearn-compat
-scipy==1.15.2
-    # via
-    #   batchalign (./setup.py)
-    #   hmmlearn
-    #   imbalanced-learn
-    #   scikit-learn
-sentencepiece==0.2.0
-    # via batchalign (./setup.py)
-setuptools==76.0.0
-    # via batchalign (./setup.py)
-six==1.17.0
-    # via
-    #   python-dateutil
-    #   rev-ai
-    #   websocket-client
-sklearn-compat==0.1.3
-    # via imbalanced-learn
-soundfile==0.12.1
-    # via batchalign (./setup.py)
-stanza==1.10.1
-    # via batchalign (./setup.py)
-sympy==1.13.1
-    # via torch
-threadpoolctl==3.6.0
-    # via
-    #   imbalanced-learn
-    #   scikit-learn
-tiktoken==0.9.0
-    # via batchalign (./setup.py)
-tokenizers==0.21.1
-    # via
-    #   batchalign (./setup.py)
-    #   transformers
-toml==0.10.2
-    # via coverage
-torch==2.6.0
-    # via
-    #   batchalign (./setup.py)
-    #   accelerate
-    #   peft
-    #   stanza
-    #   torchaudio
-torchaudio==2.6.0
-    # via batchalign (./setup.py)
-tqdm==4.67.1
-    # via
-    #   huggingface-hub
-    #   nltk
-    #   peft
-    #   stanza
-    #   transformers
-transformers==4.50.0
-    # via
-    #   batchalign (./setup.py)
-    #   peft
-    #   stanza
-typing-extensions==4.12.2
-    # via
-    #   batchalign (./setup.py)
-    #   huggingface-hub
-    #   praatio
-    #   pydantic
-    #   pydantic-core
-    #   rich-click
-    #   torch
-urllib3==2.3.0
-    # via
-    #   blobfile
-    #   requests
-websocket-client==0.59.0
-    # via rev-ai
-
-# Additional dependencies that might help with batchalign
-jiwer>=3.0.0
-textgrid>=1.5
-pympi-ling>=1.70
-
-# Utilities
-jsonlines>=3.0.0
-setuptools<81
+# --- Batchalign and friends (if you need them) ---
+praatio>=6.0
+stanza>=1.7
+hmmlearn>=0.3
+num2words>=0.5
+eyed3>=0.9
+plotly>=5.0