asoria HF staff commited on
Commit
457b45f
1 Parent(s): c3813c7

Try to avoid calling datamapplot

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -178,11 +178,12 @@ def generate_topics(dataset, config, split, column, nested_column):
178
  all_docs.extend(docs)
179
 
180
  topics_info = base_model.get_topic_info()
181
- topic_plot = base_model.visualize_documents(
182
- all_docs,
183
- reduced_embeddings=np.vstack(reduced_embeddings_list),
184
- custom_labels=True,
185
- )
 
186
 
187
  logging.info(f"Topics: {llama2_labels}")
188
 
 
178
  all_docs.extend(docs)
179
 
180
  topics_info = base_model.get_topic_info()
181
+ # topic_plot = base_model.visualize_documents(
182
+ # all_docs,
183
+ # reduced_embeddings=np.vstack(reduced_embeddings_list),
184
+ # custom_labels=True,
185
+ # )
186
+ topic_plot = base_model.visualize_barchart()
187
 
188
  logging.info(f"Topics: {llama2_labels}")
189