gradio==3.50.2
torch>=2.1.0
transformers
tokenizers
einops