Update app.py

app.py

@@ -3,16 +3,18 @@ import streamlit as st
 
 
 @st.cache()
-def load_all_repository_names():
-    list_of_repo_names = ["test"]
-    #list_of_repo_names = load_dataset("bigcode/the-stack-paths", split="train")["repository_name"]
-    return list_of_repo_names
+def load_all_usernames():
+    list_of_usernames = load_dataset("bigcode/the-stack-usernames", split="train")["usernames"]
+    return set(list_of_repo_names)
 
 st.title("Am I in The Stack?")
 st.markdown("This tool lets you check if a repository under a given username is part of [The Stack dataset](https://huggingface.co/datasets/bigcode/the-stack).")
-repo_names = load_all_repository_names()
-n_repos = len(repo_names)
+usernames = load_all_usernames()
 
 username = st.text_input("GitHub Username:")
 
-st.markdown("_Note_: this Space is currently under construction.")
+if st.button("Check!"):
+    if username in usernames:
+        st.markdown("**Yes**, your data is in The Stack.")
+    else:
+        st.markdown("**No**, your data is not in The Stack.")