torch==2.4.1
transformers==4.41.2
streamlit==1.40.0
einops