Upload start.sh with huggingface_hub

start.sh

@@ -0,0 +1,25 @@
+#!/bin/bash
+# Download data from HF Bucket on startup, then start API
+python -c "
+from huggingface_hub import download_bucket_files, list_bucket_tree
+import os
+
+BUCKET = os.environ.get('HF_BUCKET', 'stasaking/blood-brain-benchmark')
+
+files = list(list_bucket_tree(BUCKET))
+download_list = []
+for f in files:
+    if f.path.endswith(('.parquet', '.json')):
+        local_path = f'/app/data/{f.path}'
+        os.makedirs(os.path.dirname(local_path), exist_ok=True)
+        download_list.append((f.path, local_path))
+    elif f.path == 'hf-space/main.py':
+        download_list.append((f.path, '/app/main.py'))
+
+print(f'Downloading {len(download_list)} files...')
+download_bucket_files(BUCKET, files=download_list)
+print('Data downloaded')
+"
+
+export BENCHMARK_DATA_DIR=/app/data
+uvicorn main:app --host 0.0.0.0 --port 7860