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>.

</center>
</p>
"""
# description = """
# <p>
# <center>
# <style type="text/css">
#  .nobull {
#   list-style-type: none;
#  }
# </style>
# 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: 
# <ul style="nobull">
#   <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 coefficient of modification (higher value means stronger modification)</li>
#   <li>🔥 Submit!</li>
# </ul>
# 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>
# """
title = "InterfaceGAN++ Demo"