torch
transformers==4.15.0
streamlit