Sure, so mark it up, just don't grade it or count its completion in a way such that it counts toward final rankings vs other students. Offer your TA's services to those who want to use it to improve, not to evaluate their AI's output.