diff --git a/nlp/cli.py b/nlp/cli.py
index 113b0a2..02e2d28 100644
--- a/nlp/cli.py
+++ b/nlp/cli.py
@@ -17,6 +17,7 @@
 from sklearn.linear_model import LogisticRegression
 from nltk.corpus import stopwords
 from nltk.stem import PorterStemmer
+import tensorflow.keras as keras
 from tensorflow.keras.models import load_model
 from tensorflow.keras.preprocessing.text import Tokenizer
 from tensorflow.keras.preprocessing.sequence import pad_sequences