Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,6 @@ st.markdown("<h1 style='text-align: center;'>Topic Trendy</h1>", unsafe_allow_ht
|
|
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
|
23 |
-
input_text = st.text_area("Enter product topic here")
|
24 |
|
25 |
topic = pd.read_csv('./Data/tiktok_utf8.csv')
|
26 |
|
@@ -31,7 +30,10 @@ vectorizer_model = CountVectorizer(stop_words="english")
|
|
31 |
topic_model = BERTopic(verbose=True,vectorizer_model=vectorizer_model)
|
32 |
#topics, probs = topic_model.fit_transform(tiktok)
|
33 |
|
34 |
-
|
|
|
|
|
|
|
35 |
|
36 |
most_similar = similar_topics[0]
|
37 |
print(similar_topics[0])
|
|
|
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
|
|
|
23 |
|
24 |
topic = pd.read_csv('./Data/tiktok_utf8.csv')
|
25 |
|
|
|
30 |
topic_model = BERTopic(verbose=True,vectorizer_model=vectorizer_model)
|
31 |
#topics, probs = topic_model.fit_transform(tiktok)
|
32 |
|
33 |
+
placeholder = st.empty()
|
34 |
+
text_input = placeholder.text_area("Enter product topic here", height=300)
|
35 |
+
|
36 |
+
similar_topics, similarity = topic_model.find_topics(text_input , top_n=20)
|
37 |
|
38 |
most_similar = similar_topics[0]
|
39 |
print(similar_topics[0])
|