message: Now caching properly and fixed a typo (def BahdanauAttention -> class BahdanauAttention)
diff: "cell_type": "code",
  "source": [
  "# getting the unique images\n",
  - "encode_train = sorted(set(img_name_train))\n",
  + "encode_train = sorted(set(img_name_vector))\n",
  "\n",
  "# feel free to change the batch_size according to your system configuration\n",
  "image_dataset = tf.data.Dataset.from_tensor_slices(\n",
  },
  "cell_type": "code",
  "source": [
  - "def BahdanauAttention(tf.keras.Model):\n",
  + "class BahdanauAttention(tf.keras.Model):\n",
  " def __init__(self, units):\n",
  " super(BahdanauAttention, self).__init__()\n",
  " self.W1 = tf.keras.layers.Dense(units)\n",
  