torch==2.0.1
transformers
gradio
datasets