gradio
torch
transformers
matplotlib
numpy