torch
pytorch-lightning
torchmetrics
numpy
scikit-learn
gradio
huggingface-hub
matplotlib
Pillow
openai