younesbelkada commited on
Commit
7bf4347
1 Parent(s): dec5828
Files changed (2) hide show
  1. app.py +10 -3
  2. utils/constants.py +2 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from yarg import get
9
  from models.stylegan_generator import StyleGANGenerator
10
  from models.stylegan2_generator import StyleGAN2Generator
11
 
12
- from utils.constants import VALID_CHOICES, ENABLE_GPU, MODEL_NAME, OUTPUT_LIST, description
13
  from utils.image_manip import tensor_to_pil, concat_images
14
 
15
  def get_generator(model_name):
@@ -76,11 +76,18 @@ iface = gr.Interface(
76
  default=2,
77
  label="Number of images to generate",
78
  ),
79
- gr.inputs.CheckboxGroup(VALID_CHOICES, default=[], type="value", label=None, optional=False)
 
 
 
 
 
 
80
  ],
81
  outputs=OUTPUT_LIST,
82
  layout="horizontal",
83
  theme="peach",
84
- description=description
 
85
  )
86
  iface.launch()
 
9
  from models.stylegan_generator import StyleGANGenerator
10
  from models.stylegan2_generator import StyleGAN2Generator
11
 
12
+ from utils.constants import VALID_CHOICES, ENABLE_GPU, MODEL_NAME, OUTPUT_LIST, description, title
13
  from utils.image_manip import tensor_to_pil, concat_images
14
 
15
  def get_generator(model_name):
 
76
  default=2,
77
  label="Number of images to generate",
78
  ),
79
+ gr.inputs.CheckboxGroup(
80
+ VALID_CHOICES,
81
+ default=[],
82
+ type="value",
83
+ label="Select attributes to modify",
84
+ optional=False
85
+ )
86
  ],
87
  outputs=OUTPUT_LIST,
88
  layout="horizontal",
89
  theme="peach",
90
+ description=description,
91
+ title=title,
92
  )
93
  iface.launch()
utils/constants.py CHANGED
@@ -30,4 +30,5 @@ Check the <a href="https://github.com/genforce/interfacegan">original repo</a> a
30
 
31
  </center>
32
  </p>
33
- """
 
 
30
 
31
  </center>
32
  </p>
33
+ """
34
+ title = "InterfaceGAN++ Demo"