5.7.2 创建一个以类别为条件的UNet模型