requirements.txt

# Use CPU-only PyTorch wheels from the official index
--extra-index-url https://download.pytorch.org/whl/cpu
torch==2.2.2+cpu
torchvision==0.17.2+cpu
torchaudio==2.2.2+cpu
torchmetrics
torchcam
captum
# Lightning (pytorch-lightning older alias still works but this is preferred)
lightning
albumentations
# Headless OpenCV for servers without GUI
opencv-python-headless
matplotlib
seaborn
tensorboard
rich
PyYAML
streamlit
requests
gradio