Update shape.py

shape.py

@@ -3,6 +3,7 @@ from shap_e.diffusion.gaussian_diffusion import diffusion_from_config as diffusi
 from shap_e.models.download import load_model, load_config
 from shap_e.util.notebooks import create_pan_cameras, decode_latent_images, gif_widget
 from shap_e.util.notebooks import decode_latent_mesh
+import torch
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 xm = load_model('transmitter', device=device)
 model = load_model('text300M', device=device)