Upload streamlit_app.py

src/streamlit_app.py

@@ -11,13 +11,26 @@ from huggingface_hub import login, snapshot_download
 
 st.set_page_config(layout="wide")
 
+os.environ.setdefault("HF_HUB_DISABLE_PROGRESS_BARS", "1")
+
 hf_token = os.getenv("HF_TOKEN")
-if hf_token:
-    login(token=hf_token)
-else:
+if not hf_token:
     raise ValueError("environment variable HF_TOKEN not found.")
 
-repo_dir = snapshot_download('Johnny-Z/dataset_viewer', repo_type='dataset')
+@st.cache_resource
+def init_hf_repo():
+
+    login(token=hf_token)
+
+    repo_dir = snapshot_download(
+        'Johnny-Z/dataset_viewer',
+        repo_type='dataset',
+        max_workers=1,
+        local_dir_use_symlinks=False
+    )
+    return repo_dir
+
+repo_dir = init_hf_repo()
 
 data_source = st.sidebar.radio("Source", ["Danbooru"], index=0)
 if data_source == "Danbooru":