torch
pytorch
streamlit
gradio