Update prepare_data.py

Forgot to import `pickle`

Author: hwong557

code/prepare_data.py

@@ -19,6 +19,7 @@
 import numpy as np
 import random
 import imageio
+import pickle
 
 from natsort import natsort
 from tqdm import tqdm