torch
transformers
gradio==4.36.0
sentencepiece