numpy==1.21.6
python-dotenv==0.21.0
scikit_learn==0.24.2
setuptools==61.2.0
torch
torchvision
tqdm==4.64.0