start_training.sh

#!/bin/bash

# Start training and inference server for Hugging Face Space

echo "🚀 Starting Phi-3 DPO Training on BEIR Dataset"

# Check for required environment variables
if [ -z "$HF_TOKEN" ]; then
    echo "⚠️ Warning: HF_TOKEN not set. Model won't be pushed to Hub."
fi

if [ -z "$HF_USERNAME" ]; then
    echo "⚠️ Warning: HF_USERNAME not set. Using default."
    export HF_USERNAME="your-username"
fi

# Start inference server in background
echo "🌐 Starting inference server on port 5000..."
python inference_server.py &
INFERENCE_PID=$!

# Give inference server time to start
sleep 5

# Start training
echo "🎯 Starting DPO training..."
python train_dpo_hf.py

# Training finished
echo "✅ Training complete!"

# Keep inference server running
echo "📡 Inference server still running on port 5000"
echo "💡 You can test the final model via the API"

# Wait for inference server
wait $INFERENCE_PID