Update app.py

app.py

@@ -175,14 +175,22 @@ if task == "Clustering":
 
 @st.cache(allow_output_mutation=True)
 def fit_text_clf(X, y):
-    text_clf.fit(X, y)
+    text_clf = text_clf.fit(X, y)
     return text_clf
 
+@st.cache(allow_output_mutation=True) 
+def fit_cluster_clf(X):
+    cluster_clf = cluster_clf.fit(X)
+    return cluster_clf
+    
+
 if task == "Classification":
     #text_clf.fit(df[column_name], df[labels_column_name])
     text_clf = fit_text_clf(df[column_name], df[labels_column_name])
 else:
-    kmeans = cluster_clf.fit(df[column_name])
+    #kmeans = cluster_clf.fit(df[column_name])
+    cluster_clf = fit_cluster_clf(df[column_name])
+    kmeans = cluster_clf
     labels = list(cluster_clf['cluster'].labels_)
     #text_clf.fit(df[column_name], labels)
     text_clf = fit_text_clf(df[column_name], labels)