torch
transformers
datasets
sentencepiece
gradio==3.40.1