torch
transformers
gradio==3.14.0