Only use label data

[tinvietle]

I want to ask if I only use label data, which code should I remove to make the model works