requirements.txt

# ===== Core (Gradio inference app) =====
gradio>=4.32.0
transformers>=4.41.0
torch>=2.3.0  # For CUDA, install from https://pytorch.org/get-started/ per your GPU/driver
safetensors>=0.4.3
huggingface_hub>=0.23.0
numpy>=1.24.0
pillow>=10.0.0

# ===== Training & dataset pipeline (optional but handy if you also train or preprocess) =====
accelerate>=0.30.0
datasets>=2.20.0
pyarrow>=15.0.0
evaluate>=0.4.2
iterative-stratification>=0.1.7
matplotlib>=3.8.0
tensorboard>=2.16.0
opencv-python>=4.8.0
scikit-learn>=1.3.0
scipy>=1.10.0

# ===== (Optional) convenience =====
# streamlit>=1.33.0  # only if you still run a Streamlit demo