entrypoint.sh

#!/bin/bash
set -e

python -c "from huggingface_hub import hf_hub_download; \
    hf_hub_download(repo_id='existedyear/scamshield_ai', filename='best_model.pt', \
    local_dir='/app/smishing_detector', local_dir_use_symlinks=False, token=True)"

exec gunicorn -w 1 -b 0.0.0.0:7860 --timeout 120 app.flask_api:app