torch
tensorflow-gpu
transformers
gradio