# Requirements for HuggingFace Inference Endpoint
# This file contains dependencies needed for the handler.py to work

# Core ML frameworks
transformers>=4.30.0
torch>=2.0.0

# Audio processing
librosa>=0.10.0
soundfile>=0.12.0
audioread>=3.0.0

# Numerical computing
numpy>=1.24.0
scipy>=1.10.0

# Additional dependencies for SongFormer model
# (these may already be installed by the model itself)
einops>=0.7.0
x-transformers>=1.0.0
ema-pytorch>=0.2.0
loguru>=0.7.0
omegaconf>=2.3.0
muq
msaf



# Note: For MP3 support, ffmpeg must be installed in the system
# Add to Dockerfile:
# RUN apt-get update && apt-get install -y ffmpeg