torch
transformers
gradio==3.26.0