gradio
torch
transformers
scikit-learn