RamAnanth1 commited on
Commit
3957214
1 Parent(s): 2b70172

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -81,7 +81,8 @@ def get_visualizations():
81
  embeddings=embeds_npy,
82
  topics = list(range(8)),
83
  custom_labels=True)
84
- fig2 = topic_model.visualize_barchart()
 
85
  st.plotly_chart(fig1)
86
  st.plotly_chart(fig2)
87
 
 
81
  embeddings=embeds_npy,
82
  topics = list(range(8)),
83
  custom_labels=True)
84
+ topic_model.set_topic_labels(topic_names)
85
+ fig2 = topic_model.visualize_barchart(custom_labels=True))
86
  st.plotly_chart(fig1)
87
  st.plotly_chart(fig2)
88