torch
transformers==4.40.1
sentencepiece
gradio