torch
git+https://github.com/openai/shap-e.git
gradio
ipywidgets