HarryLee commited on
Commit
1d605d4
·
1 Parent(s): c45d768

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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
- similar_topics, similarity = topic_model.find_topics(input_text, top_n=20)
 
 
 
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])