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