This produces a surgical mask, trained on 197 images. This was my first attempt at something like this, it produces the mask okay however there might be some SD 1.5 weirdness unfortunately.

Feel free to download and use the dataset, i am sure this could be improved.