Intelligent-Candidate-Ranking-Model / docker-entrypoint.sh
LordofMonarchs's picture
Upload folder using huggingface_hub
c754148 verified
Raw
History Blame Contribute Delete
610 Bytes
#!/bin/bash
set -e
MODE=${1:-full}
if [ "$MODE" = "full" ]; then
echo "Running full pipeline..."
python scripts/run_full_pipeline.py --candidates "$CANDIDATES_PATH" --out "$OUT_PATH"
elif [ "$MODE" = "rank" ]; then
echo "Running ranking only..."
python src/rank.py --candidates "$CANDIDATES_PATH" --out "$OUT_PATH"
elif [ "$MODE" = "precompute" ]; then
echo "Running precompute only..."
python scripts/precompute.py --candidates "$CANDIDATES_PATH" --base-dir "$BASE_DIR"
else
echo "Unknown mode: $MODE"
echo "Usage: $0 {full|rank|precompute}"
exit 1
fi