torch
transformers
gradio==4.44.1