Merge pull request #408 from GoogleCloudPlatform/fix_classify_text_wi…

…th_bert fix imports in lab folder - classify text with BERT
GoogleCloudPlatform · Nov 2, 2023 · a602011 · a602011
2 parents 2405123 + f44bc0c
commit a602011
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/notebooks/text_models/labs/classify_text_with_bert.ipynb b/notebooks/text_models/labs/classify_text_with_bert.ipynb
@@ -58,12 +58,14 @@
    },
    "outputs": [],
    "source": [
+    "import datetime\n",
     "import shutil\n",
     "\n",
     "import matplotlib.pyplot as plt\n",
     "import tensorflow as tf\n",
     "import tensorflow_hub as hub\n",
     "import tensorflow_text as text\n",
+    "from google.cloud import aiplatform\n",
     "from official.nlp import optimization  # to create AdamW optmizer\n",
     "\n",
     "tf.get_logger().setLevel(\"ERROR\")"