transformers
torch
torchvision
streamlit