Spaces:
Runtime error
Runtime error
add title
Browse files- app.py +10 -3
- 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(
|
|
|
|
|
|
|
|
|
|
|
|
|
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"
|