torch
transformers
datasets
gradio
sentencepiece