src/requirements.in

pip-tools

torch
torchvision
torchaudio 
--extra-index-url https://download.pytorch.org/whl/cu116

pytorch-lightning
transformers
matplotlib
seaborn
numpy
pandas
icecream
jupyter
gradio