Add import keras

1 files changed, 2 insertions, 1 deletions

diff --git a/cgan.py b/cgan.py
index f3bdbe5..ad3d194 100644
--- a/cgan.py
+++ b/cgan.py
@@ -1,5 +1,6 @@
 from __future__ import print_function, division
-
+import tensorflow.keras as keras
+import tensorflow as tf
 from keras.datasets import mnist
 from keras.layers import Input, Dense, Reshape, Flatten, Dropout, multiply
 from keras.layers import BatchNormalization, Activation, Embedding, ZeroPadding2D