update

.gitattributes

@@ -8,6 +8,7 @@
 *.h5 filter=lfs diff=lfs merge=lfs -text
 *.joblib filter=lfs diff=lfs merge=lfs -text
 *.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.lz4 filter=lfs diff=lfs merge=lfs -text
 *.mlmodel filter=lfs diff=lfs merge=lfs -text
 *.model filter=lfs diff=lfs merge=lfs -text
 *.msgpack filter=lfs diff=lfs merge=lfs -text
@@ -33,3 +34,22 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+# Audio files - uncompressed
+*.pcm filter=lfs diff=lfs merge=lfs -text
+*.sam filter=lfs diff=lfs merge=lfs -text
+*.raw filter=lfs diff=lfs merge=lfs -text
+# Audio files - compressed
+*.aac filter=lfs diff=lfs merge=lfs -text
+*.flac filter=lfs diff=lfs merge=lfs -text
+*.mp3 filter=lfs diff=lfs merge=lfs -text
+*.ogg filter=lfs diff=lfs merge=lfs -text
+*.wav filter=lfs diff=lfs merge=lfs -text
+# Image files - uncompressed
+*.bmp filter=lfs diff=lfs merge=lfs -text
+*.gif filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text
+*.tiff filter=lfs diff=lfs merge=lfs -text
+# Image files - compressed
+*.jpg filter=lfs diff=lfs merge=lfs -text
+*.jpeg filter=lfs diff=lfs merge=lfs -text
+*.webp filter=lfs diff=lfs merge=lfs -text

app.py

@@ -18,6 +18,7 @@ from huggingface_hub import (
     InferenceClient,
     login,
     snapshot_download,
+    hf_hub_download
 )
 from PIL import Image
 from utils import string_to_image
@@ -37,12 +38,14 @@ session_token = os.environ.get("SessionToken")
 login(token=session_token, add_to_git_credential=True)
 
 # Using snapshot_download to handle the download and extraction
-snapshot_download(
-    repo_id='XAI/PEEB-Data',
-    repo_type='dataset',
-    local_dir='./',
-    cache_dir='./hf_cache'
-)
+# snapshot_download(
+#     repo_id='XAI/PEEB-Data',
+#     repo_type='dataset',
+#     local_dir='./',
+#     cache_dir='./hf_cache'
+# )
+zip_file_path = hf_hub_download(repo_id='XAI/PEEB-Data', repo_type='dataset', cache_dir='./hf_cache', filename="data.zip")
+
 
 with zipfile.ZipFile('./data.zip', 'r') as zip_ref:
     zip_ref.extractall("./")