requirements.txt

gradio>=4.40.0
transformers>=4.44.0
torch>=2.1.0
accelerate
scikit-learn>=1.4.0
numpy>=1.26.0
einops>=0.7.0
tqdm>=4.66.0