Update app.py

app.py

@@ -3,6 +3,7 @@ import streamlit as st
 from huggingface_hub import hf_hub_download
 import gzip
 import json
+import time
 
 
 @st.cache(allow_output_mutation=True)
@@ -21,9 +22,14 @@ st.markdown("As part of the BigCode project, we released and maintain [The Stack
 
 st.markdown("This tool lets you check if a repository under a given username is part of The Stack dataset. Would you like to have your data removed from future versions of The Stack? You can opt-out following the instructions [here](https://www.bigcode-project.org/docs/about/the-stack/#how-can-i-request-that-my-data-be-removed-from-the-stack).")
 
+
+t_load = time.time()
 usernames = load_all_usernames()
+t_load = f"{time.time()-t_load:.2f}"
 username = st.text_input("Your GitHub Username:")
 
+t_display = time.time()
+
 if username or st.button("Check!"):
     if username in usernames:
         repos = usernames[username]
@@ -32,4 +38,7 @@ if username or st.button("Check!"):
         for repo in repos:
             st.markdown(f"`{repo['repository']}`")
     else:
-        st.markdown("**No**, your code is not in The Stack.")
+        st.markdown("**No**, your code is not in The Stack.")
+        
+t_display = f"{time.time()-t_load:.2f}"
+st.markdown(f"_load: {t_load}s, display: {t_display}_")