import tensorflow as tf

        with tf.Session() as sess:
            """Model function for CNN."""
            features = tf.placeholder(
                tf.float32, shape=[None, IMAGE_SIZE * IMAGE_SIZE], name='features')
