Update app.py

app.py

@@ -4,7 +4,7 @@ import streamlit as st
 
 @st.cache()
 def get_repo_names():
-    list_of_repo_names = load_dataset("bigcode/the-stack-paths")["repository_name"]#["lvwerra/test", "lvwerra/test2", "numpy/numpy"]
+    list_of_repo_names = load_dataset("bigcode/the-stack-paths", split="train")["repository_name"]#["lvwerra/test", "lvwerra/test2", "numpy/numpy"]
     return list_of_repo_names
 
 st.title("Am I in The Stack?")