Update app.py

app.py

@@ -14,14 +14,13 @@ st.markdown("This demo allows the visualization of personal information anonymiz
     This is just an illustration of BigCode's PII pipeline and the examples and secrets are synthetic.")
 
 @st.cache()
-def load_data(language="python"):
+def load_data():
     # load dataset with modified files with: content, references and language columns
     dataset = load_dataset("data", split="train")
     return dataset 
 
 
 def get_samples_tag(dataset, tag):
-    # get samples with tag
     # add column id to be able to retrieve the sample
     tmp = dataset.add_column("index", range(len(dataset)))
     samples = tmp.filter(lambda x: "PI:" + tag.upper() in x['references'])
@@ -30,6 +29,7 @@ def get_samples_tag(dataset, tag):
 
 col1, col2 = st.columns([2, 4])
 with col1:
+    #TODO add examples in more languages
     lang = st.selectbox("Select a programming language", ["Python"])
 
 samples = load_data(language=lang.lower())