torch
transformers
numpy
streamlit