datasets
faiss-cpu
gradio
lightly
lightning
numpy
Pillow
torchvision