torch
datasets
numpy
gradio