utils/constants.py

import gradio as gr

VALID_CHOICES = [
    "Bald",
    "Young",
    "Mustache",
    "Eyeglasses",
    "Hat",
    "Smiling",
    "Gray_Hair",
]
ENABLE_GPU = False
MODEL_NAME = "stylegan_ffhq"
OUTPUT_LIST = [
    gr.outputs.Image(type="pil", label="Generated Images"),
    gr.outputs.Image(type="pil", label="Modified Images"),
]
# description = """
# This is an interactive demo of an extension of the <a href="https://genforce.github.io/interfacegan/">CVPR2020 InterfaceGAN paper</a>, by adding other attributes such as Hat, Bald, etc. to the generated images. Here is a step-by-step guide to use this interface: 
#   + 🌾 Select the Random seed you want to use to generate the images  
#   + &#128450; Select the list of attributes you want to modify (of course, you can mix several attributes)  
#   + 🛠 Select the coefficient of modification (higher value means stronger modification)  
#   + 🔥 Submit!  

# Check the <a href="https://github.com/genforce/interfacegan">original repo</a> as well as the <a href="https://github.com/younesbelkada/interfacegan">extended version of the work</a>.


# &#x2B55; This method is biased on the data it has been trained for attribute recognition. E.g. if you decide to modify the "Bald" attribute on Female faces, the method will turn it into Male faces. Future work may focus more on this direction to try to have unbiased results of the modifications.
#Check the <a href="https://github.com/genforce/interfacegan">original repo</a> as well as the <a href="https://github.com/younesbelkada/interfacegan">extended version of the work</a>.
# 
# </center>
# </p>
# """
description = """
<p>
This is an interactive demo of an extension of the <a href="https://genforce.github.io/interfacegan/">CVPR2020 InterfaceGAN paper</a>, which adds other attributes such as Hat, Bald, etc. to the generated images. Here is a step-by-step guide to use this space: 
<ul>
  <li>🌾 Select the Random seed you want to use to generate the images</li>
  <li>&#128450; Select the list of attributes you want to modify (of course, you can mix several attributes)</li>
  <li>🛠 Select the modification scale (higher value means stronger modification)</li>
  <li>🔥 Submit!</li>
</ul>
&#x2B55; This method is biased on the data it has been trained for attribute recognition. E.g. if you decide to modify the "Bald" attribute on Female faces, the method will turn it into Male faces. Future work may focus more on this direction to try to have unbiased results of the modifications.
</p>
"""
css = """
ul {
  list-style-type: none;
}

ul.no-bullets li
{
    list-style-type: none;
}
"""
title = "InterfaceGAN++ Demo"
article = "<div style='text-align: center;'><a href='https://github.com/genforce/interfacegan' target='_blank'>Original Paper</a> | <a href='https://github.com/younesbelkada/interfacegan' target='_blank'>Extended Version</a></div>"