adding counts for docs

app.py

@@ -84,12 +84,12 @@ def visualization(path_data, lang, num_docs, num_docs_for_words):
     cond = np.all(cond, axis=0)
 
     data_not_keep = data.loc[np.invert(cond)]
-    st.subheader("Filtered data")
+    st.subheader(f"Filtered data: {np.invert(cond).sum()} docs")
     st.markdown("Click on a column to sort by it, place the cursor on the text to display it.")
     st.dataframe(data_not_keep)
 
     data_keep = data.loc[cond]
-    st.subheader("Kept data")
+    st.subheader(f"Kept data: {cond.sum()} docs")
     st.markdown("Click on a column to sort by it, place the cursor on the text to display it.")
     st.dataframe(data_keep)