requirements.txt

torch>=2.1,<2.9
torchvision>=0.16,<0.18
transformers>=4.34,<5
gradio>=3.41,<6
numpy>=1.25
datasets>=2.13
accelerate