Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ st.markdown("<h1 style='text-align: center;'>Topic Trendy</h1>", unsafe_allow_ht
|
|
16 |
sentence_model = SentenceTransformer(model_name)
|
17 |
umap_model = UMAP(n_neighbors=15, n_components=2, min_dist=0.1, metric="cosine")
|
18 |
hdbscan_model = HDBSCAN(min_cluster_size=5, min_samples = 3, metric="euclidean", prediction_data=True)
|
19 |
-
vectorizer_model = CountVectorizer(lowercase = True, ngram_range=(1, 3), analyzer="word", max_df=1.0, min_df
|
20 |
|
21 |
kw_model = BERTopic(embedding_model=sentence_model, umap_model = umap_model, hdbscan_model = hdbscan_model, vectorizer_model = vectorizer_model, nr_topics = "auto", calculate_probabilities = True)
|
22 |
BerTopic_model = kw_model
|
|
|
16 |
sentence_model = SentenceTransformer(model_name)
|
17 |
umap_model = UMAP(n_neighbors=15, n_components=2, min_dist=0.1, metric="cosine")
|
18 |
hdbscan_model = HDBSCAN(min_cluster_size=5, min_samples = 3, metric="euclidean", prediction_data=True)
|
19 |
+
vectorizer_model = CountVectorizer(lowercase = True, ngram_range=(1, 3), analyzer="word", max_df=1.0, min_df=0.5 stop_words="english")
|
20 |
|
21 |
kw_model = BERTopic(embedding_model=sentence_model, umap_model = umap_model, hdbscan_model = hdbscan_model, vectorizer_model = vectorizer_model, nr_topics = "auto", calculate_probabilities = True)
|
22 |
BerTopic_model = kw_model
|