:art: add progress

app.py

@@ -19,12 +19,19 @@ def run_cmd(command):
         sys.exit(1)
 
 
-def download_cache(cache_dir: str, repo_name: str):
+def download_cache(cache_dir: str, repo_name: str, wait_for_completion: bool = True):
     """Clones a repo from HuggingFace Hub to a cache directory"""
     if os.environ.get("DO_DOWNLOAD_CACHE") and 'cache_is_downloaded' not in st.session_state:
         with st.spinner("Downloading cache...this might take a while 😬"):
             repo = Repository(local_dir=cache_dir, clone_from=repo_name, repo_type='dataset')
             repo.git_pull()
+            if wait_for_completion:
+                placeholder = st.empty()
+                elapsed = 0
+                while os.environ.get("GIT_LFS_PROGRESS"):
+                    time.sleep(1)
+                    elapsed += 1
+                    placeholder.write(f"Been waiting for {elapsed}s")
 
 
 def main():