diff --git a/avocado-training.py b/avocado-training.py
index 2211db7..1f71295 100644
--- a/avocado-training.py
+++ b/avocado-training.py
@@ -5,6 +5,7 @@ from sklearn import preprocessing
 from sklearn.linear_model import LinearRegression
 from sklearn.metrics import mean_squared_error
 import tensorflow as tf
+from tensorflow import keras
 from tensorflow.keras.layers import Input, Dense, Activation,Dropout
 from tensorflow.keras.models import Model
 from tensorflow.keras.callbacks import EarlyStopping
diff --git a/requirements.txt b/requirements.txt
index 454e683..f9d5201 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,5 +3,6 @@ pandas==1.2.4
 numpy==1.19.2
 sklearn
 tensorflow==2.4.1
+keras
 jinja2==2.11.3
 matplotlib
\ No newline at end of file