Quantum-aware Transformer model for state classification