requirements.txt

albumentations==1.3.0
gradio==3.32.0
numpy==1.22.4
opencv_python==4.7.0.68
scikit_learn==1.2.2
torch==1.13.0
torchvision==0.14.0
tqdm==4.64.1