rescored / backend /scripts /test_mps_performance.sh
calebhan's picture
yourmt3 integration and refactor
75d3906
#!/bin/bash
# Test MPS performance with optimizations
echo "====================================="
echo "YourMT3+ MPS Performance Test"
echo "====================================="
echo ""
# Start service in background
echo "Starting transcription service with MPS + float16..."
cd /Users/calebhan/Documents/Coding/Personal/rescored/backend/transcription-service
source ../backend/.venv/bin/activate 2>/dev/null || true
python service.py > service.log 2>&1 &
SERVICE_PID=$!
echo "Service PID: $SERVICE_PID"
echo "Waiting for service to initialize (30s)..."
sleep 30
# Check health
echo ""
echo "Checking service health..."
curl -s http://localhost:8001/health | python -m json.tool
# Run test transcription with timing
echo ""
echo "Running test transcription..."
echo "Audio file: ../../audio.wav"
echo ""
START_TIME=$(date +%s)
curl -X POST "http://localhost:8001/transcribe" \
-F "file=@../../audio.wav" \
--output test_mps_output.mid \
--max-time 600
END_TIME=$(date +%s)
ELAPSED=$((END_TIME - START_TIME))
echo ""
echo "====================================="
echo "Results:"
echo "====================================="
echo "Processing time: ${ELAPSED}s"
echo "MIDI output size: $(ls -lh test_mps_output.mid 2>/dev/null | awk '{print $5}')"
echo ""
echo "Service log (last 20 lines):"
tail -20 service.log
echo ""
echo "====================================="
# Cleanup
echo "Stopping service (PID: $SERVICE_PID)..."
kill $SERVICE_PID 2>/dev/null || true
echo "Done!"