xqtl-api / start.sh
stasaking's picture
Upload start.sh with huggingface_hub
32da843 verified
Raw
History Blame Contribute Delete
858 Bytes
#!/bin/bash
# Download data + code from HF Bucket on startup
python -c "
from huggingface_hub import download_bucket_files, list_bucket_tree
import os
# List all files in the bucket
files = list(list_bucket_tree('stasaking/xqtl-data'))
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('stasaking/xqtl-data', files=download_list)
print('Data downloaded')
"
# Start API — data lives in /app/data, not relative to main.py
export XQTL_DATA_DIR=/app/data
uvicorn main:app --host 0.0.0.0 --port 7860