torch
numpy
transformers
streamlit