Update utils/common_viz.py

utils/common_viz.py

@@ -4,6 +4,7 @@ import clip
 from hydra import compose, initialize
 from hydra.utils import instantiate
 from omegaconf import OmegaConf
+import spaces
 import torch
 from torchtyping import TensorType
 from torch.utils.data import DataLoader