torch
transformers
streamlit