HarryLee commited on
Commit
39a2b1e
·
1 Parent(s): b326886

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ st.set_page_config(page_title='eRupt Topic Trendy (e-Commerce x Social Media)',
13
  st.markdown("<h1 style='text-align: center;'>Topic Trendy</h1>", unsafe_allow_html=True)
14
  #BerTopic_model = BERTopic.load("my_topics_model")
15
 
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")
 
13
  st.markdown("<h1 style='text-align: center;'>Topic Trendy</h1>", unsafe_allow_html=True)
14
  #BerTopic_model = BERTopic.load("my_topics_model")
15
 
16
+ sentence_model = SentenceTransformer("all-MiniLM-L6-v2")
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")