Update app.py

Try caching more stuff

app.py

@@ -122,14 +122,19 @@ if task == "Clustering":
         ('vect', embedder),
         ('cluster', KMeans(n_clusters = n_clusters, n_init = n_init, max_iter = max_iter)),
     ])
-    
+
+@st.cache
+def fit_text_clf(X, y):
+    text_clf.fit(X, y)
 
 if task == "Classification":
-    text_clf.fit(df[column_name], df[labels_column_name])
+    #text_clf.fit(df[column_name], df[labels_column_name])
+    fit_text_clf(df[column_name], df[labels_column_name])
 else:
     kmeans = cluster_clf.fit(df[column_name])
     labels = list(cluster_clf['cluster'].labels_)
-    text_clf.fit(df[column_name], labels)
+    #text_clf.fit(df[column_name], labels)
+    fit_text_clf(df[column_name], labels)
     st.write("Generated Clusters for each example")
     st.write(labels)
 
@@ -190,8 +195,8 @@ html = format_as_html(t_pred)
 
 form_html = st.sidebar.form("html_size_form")
 form_html.header("Model Explanation Display Settings")
-output_width = form_html.number_input("Enter the number of pixels for width of model explanation html display", value = 4000)
-output_height = form_html.number_input("Enter the number of pixels for height of model explanation html display", value = 4000)
+output_width = form_html.number_input("Enter the number of pixels for width of model explanation html display", value = 2000)
+output_height = form_html.number_input("Enter the number of pixels for height of model explanation html display", value = 2000)
 form_html.form_submit_button("Submit")
 st.caption("Scroll to see the full output!")
 components.html(html, width = output_width, height = output_height, scrolling = True)