Update app.py

app.py

@@ -28,7 +28,13 @@ tiktok = topic.text.to_list()
 
 vectorizer_model = CountVectorizer(stop_words="english")
 topic_model = BERTopic(verbose=True,vectorizer_model=vectorizer_model)
-topics, probs = topic_model.fit_transform(tiktok)
+
+@st.cache()
+def fit_transform(model, docs):
+    topics, probs = model.fit_transform(docs)
+    return topics, probs
+
+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)