Update app.py

app.py

@@ -126,15 +126,16 @@ if task == "Clustering":
 @st.cache
 def fit_text_clf(X, y):
     text_clf.fit(X, y)
+    return text_clf
 
 if task == "Classification":
     #text_clf.fit(df[column_name], df[labels_column_name])
-    fit_text_clf(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])
     labels = list(cluster_clf['cluster'].labels_)
     #text_clf.fit(df[column_name], labels)
-    fit_text_clf(df[column_name], labels)
+    text_clf = fit_text_clf(df[column_name], labels)
     st.write("Generated Clusters for each example")
     st.write(labels)