deepfake-server / huggingface /sync_bundle.sh
DevQueen's picture
Sync from GitHub via hub-sync
1dc2504 verified
Raw
History Blame Contribute Delete
1.06 kB
#!/usr/bin/env bash
# Copy runtime files into huggingface/ when using Space root directory = huggingface/
set -euo pipefail
ROOT="$(cd "$(dirname "$0")/.." && pwd)"
HF="$(cd "$(dirname "$0")" && pwd)"
echo "Syncing Hugging Face bundle into ${HF}..."
if [[ ! -f "${ROOT}/outputs/best.pt" ]]; then
echo "ERROR: outputs/best.pt not found. Train epoch 1 first."
exit 1
fi
rm -rf "${HF}/api" "${HF}/src" "${HF}/configs" "${HF}/outputs"
mkdir -p "${HF}/outputs"
cp -R "${ROOT}/api" "${HF}/api"
cp -R "${ROOT}/src" "${HF}/src"
cp -R "${ROOT}/configs" "${HF}/configs"
cp "${ROOT}/outputs/best.pt" "${HF}/outputs/best.pt"
rm -f "${HF}/src/data/stream_ff_dataset.py" \
"${HF}/src/data/build_metadata.py" \
"${HF}/src/data/extract_frames.py" \
"${HF}/src/data/extract_eye_sequences.py" \
"${HF}/src/train/train.py" \
"${HF}/src/train/adversarial.py" 2>/dev/null || true
rm -rf "${HF}/src/eval" "${HF}/src/train" "${HF}/src/viz" 2>/dev/null || true
echo "Done. Commit huggingface/api, src, configs, outputs if using subfolder deploy."