diff --git a/build_map.py b/build_map.py
index 400dbb9b..53741077 100644
--- a/build_map.py
+++ b/build_map.py
@@ -49,4 +49,6 @@ atlas.map_embeddings(embeddings,
                      id_field="index",
                      name="Post Epoch 1 Embeddings",
                      colorable_fields=["source", "loss", "trained_on"],
+                     build_topic_model=True,
+                     topic_label_field="inputs",
                      reset_project_if_exists=True,)
\ No newline at end of file