numpy
torch
streamlit