Update app.py

app.py

@@ -34,7 +34,7 @@ def fit_transform(model, docs):
     topics, probs = model.fit_transform(docs)
     return topics, probs
 
-topics, probs = fit_transform(topic_model, tiktok)
+#topics, probs = fit_transform(topic_model, tiktok)
 
 #topics_over_times = topic_model.topics_over_time(tiktok, topics, timestamps, nr_bins=20)
 #topic_model.visualize_topics_over_time(topics_over_times, top_n_topics=30)
@@ -48,21 +48,27 @@ form = st.sidebar.form("Main Settings")
 form.header("Main Settings")
 
 ebay_topic= form.selectbox("eBay Products Topic Selection", ["Motor", "Bicycle", "Beauty", "Basketball", "Fitness"])
-top_n = st.sidebar.slider("Select a number of topics", 1, 30, 5, 1)
+top_n = form.number_input("What's the max length of the text?", value = 10)
 
 form.form_submit_button("Run")
 
 if ebay_topic == "Motor":
+    topics, probs = fit_transform(topic_model, tiktok)
     similar_topics, similarity = topic_model.find_topics("Motor", top_n=top_n)
 elif ebay_topic == "Bicycle":
+    topics, probs = fit_transform(topic_model, tiktok)
     similar_topics, similarity = topic_model.find_topics("Bicycle", top_n=top_n)
 elif ebay_topic == "Beauty":
+    topics, probs = fit_transform(topic_model, tiktok)
     similar_topics, similarity = topic_model.find_topics("Beauty", top_n=top_n)
 elif ebay_topic == "Basketball":
+    topics, probs = fit_transform(topic_model, tiktok)
     similar_topics, similarity = topic_model.find_topics("Basketball", top_n=top_n)
 elif ebay_topic == "Fitness":
+    topics, probs = fit_transform(topic_model, tiktok)
     similar_topics, similarity = topic_model.find_topics("Fitness", top_n=top_n)
 else:
+    topics, probs = fit_transform(topic_model, tiktok)
     similar_topics, similarity = topic_model.find_topics("Motor", top_n=top_n)
 
 if similar_topics != []: