Update app.py

app.py

@@ -242,8 +242,11 @@ def extract_problem_domains(df, text_column='Problem_Description', cluster_range
 
 
     # BERTopic for topic modelling
-    topic_model = BERTopic(num_topics=optimal_n_clusters)
+    # topic_model = BERTopic(num_topics=optimal_n_clusters)
+    # topics, _ = topic_model.fit_transform(df[text_column].tolist())
+    topic_model = BERTopic()
     topics, _ = topic_model.fit_transform(df[text_column].tolist())
+    topic_model.reduce_topics(df[text_column].tolist(), nr_topics=optimal_n_clusters)
 
 
     # Get representative words for each cluster