Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import huggingface_hub as hf_hub
 import monai
 import os
+import zipfile
 
 hf_hub.login(token=st.secrets["HF_TOKEN"])
 
@@ -9,10 +10,9 @@ with st.spinner("Downloading Dataset"):
     data_path = hf_hub.hf_hub_download(repo_id="osbm/prostate158", filename="data.zip", repo_type="dataset")
 
 st.write(data_path)
-
-import zipfile
-with zipfile.ZipFile(data_path, 'r') as zip_ref:
-    zip_ref.extractall(".")
+with st.spinnet("Unzipping..."):
+    with zipfile.ZipFile(data_path, 'r') as zip_ref:
+        zip_ref.extractall(".")
 
 
 st.write(os.listdir(os.getcwd()))