This is an implementation of a custom model Baby-pics as a Cog model. Inspired by Yifan Goh and his webapp OurBabyAI
Follow the model pushing guide to push your own model to Replicate.
Run a prediction:
cog predict -i gender="girl" -i image=@mom2.png -i image2=@dad2.png
Output Example: