myanmar-ghost / deploy_model.sh
amkyawdev's picture
Add scripts
7dcec51 verified
Raw
History Blame Contribute Delete
789 Bytes
#!/bin/bash
# Deploy Myanmar Ghost model
set -e
echo "🚀 Starting model deployment..."
# Parse arguments
MODEL_PATH=${1:-"outputs/models/best_model.pt"}
OUTPUT_DIR=${2:-"outputs/deployment"}
FORMAT=${3:-"safetensors"}
HUB_REPO=${4:-""}
# Activate virtual environment if exists
if [ -d "venv" ]; then
source venv/bin/activate
fi
# Install deployment dependencies
pip install -q safetensors huggingface_hub
# Run deployment pipeline
python -m src.pipelines.deployment_pipeline \
--model_path "$MODEL_PATH" \
--output_dir "$OUTPUT_DIR" \
--format "$FORMAT" \
${HUB_REPO:+--push_hub --repo_id "$HUB_REPO"}
echo "✅ Deployment complete!"
echo "Model exported to: $OUTPUT_DIR"
if [ -n "$HUB_REPO" ]; then
echo "Pushed to: https://huggingface.co/$HUB_REPO"
fi