torch
streamlit
matplotlib