# torch/torchvision/torchaudio are installed by Dockerfile with CPU wheel
transformers>=5.0.0
sentencepiece          # ProtT5 tokenizer backend

# Inference
xgboost==2.0.3
scikit-learn
joblib

# Gradio demo
gradio>=4.0.0
numpy
scipy
matplotlib