requirements.txt

albumentations==1.3.0
click==8.0.4
gradio==3.36.1
huggingface_hub==0.14.1
matplotlib==3.6.0
numpy==1.23.4
pandas==1.5.1
Pillow==9.3.0
pytorch_lightning==2.0.2
scikit_learn==1.1.3
torch==1.12.0
torchvision==0.13.0
tqdm==4.64.1