torch
transformers
gradio
lightning
numpy
matplotlib
huggingface_hub
omegaconf