torch
torchvision
ipykernel
matplotlib
accelerate
segmentation-models-pytorch
gdown
tqdm
torchmetrics[image]
wandb
numpy
python-dotenv
boto3
gradio