torch
transformers
gradio
tqdm
torchvision
tiktoken