requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.2.0
torchvision==0.17.0
torchaudio==2.2.0
matplotlib==3.9.2
numpy==1.26.4
pytorch-lightning==2.4.0
scikit-learn==1.0.2
gradio==5.5.0