fix style errors (pep8)
@@ -2,7 +2,8 @@ import logging
import os
import numpy as np
-#This variable needs to be set before the first import since some warnings are raised on the first import.
+# In order not to raise tensorflow warnings, we need to set this environment
+# variable before importing the `tensorflow` package
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
import tensorflow as tf
import tensorflow.keras.preprocessing.text
@@ -10,8 +11,6 @@ import tensorflow_hub as hub
import tensorflow_text
-
-
def build_embedding_model():
""" Build model by stacking up a preprocessing layer
and an encoding layer.
