Streamlit Demo: The Controllable GAN Face Generator

This project highlights Streamlit's new hash_func feature with an app that calls on TensorFlow to generate photorealistic faces, using Nvidia's Progressive Growing of GANs and Shaobo Guan's Transparent Latent-space GAN method for tuning the output face's characteristics. For more information, check out the tutorial on Towards Data Science.

The Streamlit app is implemented in only 150 lines of Python and demonstrates the wide new range of objects that can be used safely and efficiently in Streamlit apps with hash_func.

How to run this demo

The demo requires Python 3.6 or 3.7 (The version of TensorFlow we specify in requirements.txt is not supported in Python 3.8+). We suggest creating a new virtual environment, then running:

git clone https://github.com/streamlit/demo-face-gan.git
cd demo-face-gan
pip install -r requirements.txt
streamlit run streamlit_app.py

Model Bias

Playing with the sliders, you will find biases that exist in this model. For example, moving the Smiling slider can turn a face from masculine to feminine or from lighter skin to darker. Apps like these that allow you to visually inspect model inputs help you find these biases so you can address them in your model before it's put into production.

Questions? Comments?

Please ask in the Streamlit community or check out our article.