torch==2.1.0
transformers
gradio