diff --git a/face_Recognition.py b/face_Recognition.py
index 52f3869..d66d22c 100644
--- a/face_Recognition.py
+++ b/face_Recognition.py
@@ -79,8 +79,8 @@
                          nb_val_samples = 2000)'''
 
 # fit the model
-r = model.fit_generator(
-  training_set,
+r = model.fit(
+  x=training_set,
   validation_data=test_set,
   epochs=5,
   steps_per_epoch=len(training_set),
@@ -94,8 +94,8 @@
 plt.savefig('LossVal_loss')
 
 # accuracies
-plt.plot(r.history['acc'], label='train acc')
-plt.plot(r.history['val_acc'], label='val acc')
+plt.plot(r.history['accuracy'], label='train acc')
+plt.plot(r.history['val_accuracy'], label='val acc')
 plt.legend()
 plt.show()
 plt.savefig('AccVal_acc')