| # Deep Learning Framework | |
| # PyPI ships the CUDA-enabled build. On a CPU-only machine you can save ~2.5 GB: | |
| # pip install torch --index-url https://download.pytorch.org/whl/cpu | |
| # (run that first, then pip install -r requirements.txt) | |
| torch>=2.0.0 | |
| # Web Interface | |
| gradio==6.9.0 # pinned for reproducible builds (uses gr.Timer, launch(theme=, css=, head=)) | |
| # Scientific Computing | |
| numpy>=1.24.0 | |
| scipy>=1.10.0 | |
| pandas>=2.0.0 # imported directly by app.py | |
| # Visualization | |
| matplotlib>=3.7.0 | |
| plotly>=5.15.0 | |
| Pillow>=10.0.0 | |
| # Model backbones | |
| einops>=0.7.0 # required by models/vit.py | |