gradio Pillow trimesh scipy torchvision # Added, as it's used for image transforms shap-e @ git+https://github.com/openai/shap-e.git torch>=2.1.0 # Or specify CUDA-specific version like torch==2.3.0+cu118 torchaudio>=2.1.0 torchvision>=0.16.0 diffusers transformers accelerate