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])
|