import gradio as gr

VALID_CHOICES = [
    "Bald",
    "Young",
    "Mustache",
    "Eyeglasses",
    "Hat",
    "Smiling",
    "Gender",
    "Beard"
]
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 = """
<p>
<center>
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 class="list-unstyled">
  <li style="list-style-type: none;">🌾 Select the Random seed you want to use to generate the images</li>
  <li style="list-style-type: none;">&#128450; Select the list of attributes you want to modify (of course, you can mix several attributes)</li>
  <li style="list-style-type: none;">🛠 Select the coefficient of modification (higher value means stronger modification)</li>
  <li style="list-style-type: none;">🔥 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>
"""