utils/constants.py

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 CVPR2020 InterfaceGAN paper, 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>
  <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>🔥 Submit!</li>
</ul>


</center>
</p>
"""