Update demo.py

demo.py

@@ -64,8 +64,7 @@ with tf.Session() as session:
 
 
     # train & validate. Allocate 20% for validation set. 100 epochs
-    history = model.fit(np.array(data_train), np.array(labels_train), batch_size = 32, epochs=100, callbacks=[early_stopping_
-callback], verbose=1, shuffle=True, validation_split=0.2)
+    history = model.fit(np.array(data_train), np.array(labels_train), batch_size = 32, epochs=100, callbacks=[early_stopping_callback], verbose=1, shuffle=True, validation_split=0.2)
     # plot
     plt.plot(history.history['loss'])
     plt.plot(history.history['val_loss'])