streamlit
torch
torch-scatter
transformers
tensorflow==2.7.0
tensorflow_probability