Update app.py

app.py

@@ -14,7 +14,7 @@ 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():
+def load_data(language="python"):
     # load dataset with modified files with: content, references and language columns
     dataset = load_dataset("data", split="train")
     return dataset