stylegan-human / StyleGAN-Human

StyleGAN-Human: A Data-Centric Odyssey of Human Generation
1.15k stars 144 forks source link

docker container/flask api prediction for style-mixing via cog #12

Open afiaka87 opened 2 years ago

afiaka87 commented 2 years ago

Hi - i work with replicate.com. This model is really impressive! I added support for cog, which will sets up a docker container with the model initialized and exposes a restful api to grab predictions from.

I added a demo for style-mixing to replicate as well, you can claim it with this url.

Let me know if you'd like any changes in the PR :)