gradio
torch
torchvision
matplotlib
Pillow
joblib
huggingface_hub
lightning