torch
transformers
gradio==3.0.18