diff --git a/run_models.py b/run_models.py
index c13cd9e..1d7c7fd 100644
--- a/run_models.py
+++ b/run_models.py
@@ -10,7 +10,6 @@
 
 from IPython import display
 
-import imageio
 import pickle
 import os
 import time
@@ -24,7 +23,6 @@
 from polar import *
 from pac_code import *
 
-from sklearn.manifold import TSNE
 import math
 import random
 import numpy as np