update

app.py

@@ -26,7 +26,7 @@ import matplotlib.backends.backend_agg as agg
 import math
 from pathlib import Path
 import zipfile
-
+import gdown
 
 np.random.seed(int(time.time()))
 csv.field_size_limit(sys.maxsize)
@@ -37,31 +37,15 @@ np.random.seed(int(time.time()))
 session_token = os.environ.get("SessionToken")
 login(token=session_token, add_to_git_credential=True)
 
-zip_file_path = hf_hub_download(
-    repo_id="XAI/PEEB-Data",
-    repo_type="dataset",
-    cache_dir="./hf_cache",
-    filename="data.zip",
-)
-
-print(f"zip_file_path: {zip_file_path}")
-
-if os.path.exists(zip_file_path):
-    print(f"File size: {os.path.getsize(zip_file_path)} bytes")
-else:
-    print("File does not exist.")
-
-try:
-    with zipfile.ZipFile(zip_file_path, "r") as zip_ref:
-        zip_ref.extractall("./")
-        print("Extraction successful.")
-except Exception as e:
-    print(f"Failed to extract the zip file: {e}")
-
-import os
+zip_file_path = './dogs.zip'
 
-# After the extraction line
-print("Directory contents:", os.listdir("./dogs/topK/"))
+# Download Embeddings
+gdown.cached_download(
+    url="https://huggingface.co/datasets/XAI/PEEB-Data/resolve/main/data.zip?download=true",
+    path="zip_file_path",
+    quiet=False,
+    md5="153c2a3a8bf77a075f8254e191009772",
+)
 
 try:
     with zipfile.ZipFile(zip_file_path, "r") as zip_ref:

requirements.txt

@@ -0,0 +1 @@
+gdown