numpy
torch>=2.0.0
tqdm
termcolor
streamlit_extras