--extra-index-url https://download.pytorch.org/whl/cu117
torch==2.0.0
torchvision==0.15.0
torch_kmeans 
numpy==1.26.4
imageio
gradio
timm==0.9.10
tqdm
scikit-learn
pillow
matplotlib
albumentations==1.3.1