from paintbyword import StyleganPainter
from paintbyword.utils import show, pilim
# Create instance of painter class
painter = StyleganPainter(pretrained='birds')
seed_index = 0
z, image, loss_history, im_history = painter.paint(
  seed_index,
  'A photo of a yellow bird with black colored wings',
  optim_method='cma + adam'
)
# Show the painted output image.
show(pilim(image[0]))

from paintbyword import StyleganMaskedPainter
from paintbyword.utils import show, pilim
# Create instance of masked painter class
painter = StyleganMaskedPainter(pretrained='bedroom')
# Display grid of seed images to choose from
painter.show_seed_images(batch_size=32)
choice = 30  # Choose index of desired seed image.
painter.mask_seed_image(choice)  # Scribble mask on seed image.

result = painter.paint(
    choice,
    description='A photo of a rustic bed',
    optim_method='cma + adam'
)
# Show the painted output image.
show(pilim(image[0]))