Spaces:

SantanuBanerjee
/

TaxDirection

Sleeping

SantanuBanerjee commited on Aug 4, 2024

Commit

ae6ae7c

verified ·

1 Parent(s): 8b497ae

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -242,7 +242,7 @@ def extract_problem_domains(df, text_column='Problem_Description', cluster_range
     # BERTopic for topic modelling
-    topic_model = BERTopic(n_topics=optimal_n_clusters)
     topics, _ = topic_model.fit_transform(df[text_column].tolist())

     # BERTopic for topic modelling
+    topic_model = BERTopic(num_topics=optimal_n_clusters)
     topics, _ = topic_model.fit_transform(df[text_column].tolist())