Update app.py

app.py

@@ -71,12 +71,14 @@ app = Topically(os.environ["CO_API_KEY"])
 
 df['topic_name'], topic_names = app.name_topics((df['content.title'], df['topic']), num_generations=5)
 
-st.write("Topics extracted are:", topic_names)
+#st.write("Topics extracted are:", topic_names)
 
 topic_model.set_topic_labels(topic_names)
-# topic_model.visualize_documents(df['content.title'].values, 
-#                                 embeddings=embeds_npy,
-#                                 topics = list(range(8)),
-#                                 custom_labels=True,
-#                                 width=900,
-#                                 height=600)
+fig = topic_model.visualize_documents(df['content.title'].values, 
+                                embeddings=embeds_npy,
+                                topics = list(range(8)),
+                                custom_labels=True,
+                                width=900,
+                                height=600)
+st.pyplot(fig)
+