diff --git a/prototorch/models/vis.py b/prototorch/models/vis.py
index a5dbb59..c1e3b61 100644
--- a/prototorch/models/vis.py
+++ b/prototorch/models/vis.py
@@ -5,11 +5,6 @@ import pytorch_lightning as pl
 import torch
 import torchvision
 from matplotlib import pyplot as plt
-from matplotlib.offsetbox import AnchoredText
-from prototorch.utils.celluloid import Camera
-from prototorch.utils.colors import color_scheme
-from prototorch.utils.utils import (gif_from_dir, make_directory,
-                                    prettify_string)
 from torch.utils.data import DataLoader, Dataset