requirements.txt

# Core AI/ML libraries
torch==2.5.1 # Updated to meet whisperx's minimum requirement
torchaudio==2.5.1 # Updated to match torch version
transformers
accelerate
sentencepiece

# Audio processing & Diarization
pyannote.audio
soundfile
ffmpeg-python

# Gradio for the UI
gradio

# WhisperX specific
whisperx @ git+https://github.com/m-bain/whisperX.git

# General utilities (often useful)
numpy
scipy