add warning message

app.py

@@ -54,6 +54,15 @@ class Visualization:
             lang_dataset_id, path_kenlm_model
         )
 
+    def warning_preamble(self):
+        st.markdown(
+            "This demo can be a little slow, and only allows you to process up to 5000 documents "
+            "for a decent speed. If you want to display up to three times more documents and have "
+            "a faster visualization, we invite you to run this "
+            "[code](https://github.com/bigscience-workshop/data_tooling/tree/master/ac_dc/visualization) "
+            "on your computer."
+        )
+
     def preamble(self):
         st.markdown(
             "Before diving into this demo, you might want to take a look at how the filtering pipeline looks like in more detail."
@@ -170,7 +179,9 @@ class Visualization:
                 )
                 self.docs = self.docs_checkpoint
                 for i in range(len(self.docs["repetitions_ratio"])):
-                    self.docs["repetitions_ratio"].iloc[i] = self.docs["repetitions_ratio"].iloc[i][repetitions_length]
+                    self.docs["repetitions_ratio"].iloc[i] = self.docs[
+                        "repetitions_ratio"
+                    ].iloc[i][repetitions_length]
 
                 cutoff_def = "If the repetitions ratio of a document is higher than this number, the document is removed."
                 cutoff_repetitions_ratio = st.sidebar.slider(
@@ -526,6 +537,7 @@ class Visualization:
             )
 
     def visualization(self):
+        self.warning_preamble()
         self.preamble()
         self.open_data()
         self.set_title()