Python Tensorflow .predict Option