flask torch torchvision pillow faiss-cpu # or faiss-gpu if you have GPU and need it numpy