Spaces:
Runtime error
Runtime error
try with html to fix links
Browse files- app.py +5 -1
- utils/constants.py +34 -28
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, title
|
13 |
from utils.image_manip import tensor_to_pil, concat_images
|
14 |
|
15 |
def get_generator(model_name):
|
@@ -52,6 +52,8 @@ def inference(seed, coef, nb_images, list_choices):
|
|
52 |
|
53 |
return concatenated_output
|
54 |
|
|
|
|
|
55 |
iface = gr.Interface(
|
56 |
fn=inference,
|
57 |
inputs=[
|
@@ -89,5 +91,7 @@ iface = gr.Interface(
|
|
89 |
theme="peach",
|
90 |
description=description,
|
91 |
title=title,
|
|
|
|
|
92 |
)
|
93 |
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, css, article
|
13 |
from utils.image_manip import tensor_to_pil, concat_images
|
14 |
|
15 |
def get_generator(model_name):
|
|
|
52 |
|
53 |
return concatenated_output
|
54 |
|
55 |
+
# https://huggingface.co/spaces/osanseviero/6DRepNet/blob/main/app.py
|
56 |
+
|
57 |
iface = gr.Interface(
|
58 |
fn=inference,
|
59 |
inputs=[
|
|
|
91 |
theme="peach",
|
92 |
description=description,
|
93 |
title=title,
|
94 |
+
css=css,
|
95 |
+
article=article
|
96 |
)
|
97 |
iface.launch()
|
utils/constants.py
CHANGED
@@ -15,39 +15,45 @@ OUTPUT_LIST = [
|
|
15 |
gr.outputs.Image(type="pil", label="Generated Images"),
|
16 |
gr.outputs.Image(type="pil", label="Modified Images"),
|
17 |
]
|
18 |
-
description = """
|
19 |
-
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:
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
|
|
|
|
24 |
|
25 |
-
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>.
|
26 |
|
|
|
27 |
|
28 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
29 |
|
30 |
</center>
|
31 |
</p>
|
32 |
"""
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
# .nobull {
|
38 |
-
# list-style-type: none;
|
39 |
-
# }
|
40 |
-
# </style>
|
41 |
-
# 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:
|
42 |
-
# <ul style="nobull">
|
43 |
-
# <li>πΎ Select the Random seed you want to use to generate the images</li>
|
44 |
-
# <li>🗂 Select the list of attributes you want to modify (of course, you can mix several attributes)</li>
|
45 |
-
# <li>π Select the coefficient of modification (higher value means stronger modification)</li>
|
46 |
-
# <li>π₯ Submit!</li>
|
47 |
-
# </ul>
|
48 |
-
# 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>.
|
49 |
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
|
|
|
|
|
|
|
15 |
gr.outputs.Image(type="pil", label="Generated Images"),
|
16 |
gr.outputs.Image(type="pil", label="Modified Images"),
|
17 |
]
|
18 |
+
# description = """
|
19 |
+
# 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:
|
20 |
+
# + πΎ Select the Random seed you want to use to generate the images
|
21 |
+
# + 🗂 Select the list of attributes you want to modify (of course, you can mix several attributes)
|
22 |
+
# + π Select the coefficient of modification (higher value means stronger modification)
|
23 |
+
# + π₯ Submit!
|
24 |
+
|
25 |
+
# 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>.
|
26 |
|
|
|
27 |
|
28 |
+
# ⭕ This method is biased on the data it has been trained for attribute recognition. E.g. if you decide to modify the "Bald" attribute on Female faces, the method will turn it into Male faces. Future work may focus more on this direction to try to have unbiased results of the modifications.
|
29 |
|
30 |
+
# </center>
|
31 |
+
# </p>
|
32 |
+
# """
|
33 |
+
description = """
|
34 |
+
<p>
|
35 |
+
<center>
|
36 |
+
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:
|
37 |
+
<ul>
|
38 |
+
<li>πΎ Select the Random seed you want to use to generate the images</li>
|
39 |
+
<li>🗂 Select the list of attributes you want to modify (of course, you can mix several attributes)</li>
|
40 |
+
<li>π Select the coefficient of modification (higher value means stronger modification)</li>
|
41 |
+
<li>π₯ Submit!</li>
|
42 |
+
</ul>
|
43 |
+
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>.
|
44 |
|
45 |
</center>
|
46 |
</p>
|
47 |
"""
|
48 |
+
css = """
|
49 |
+
ul {
|
50 |
+
list-style-type: none;
|
51 |
+
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
52 |
|
53 |
+
ul.no-bullets li
|
54 |
+
{
|
55 |
+
list-style-type: none;
|
56 |
+
}
|
57 |
+
"""
|
58 |
+
title = "InterfaceGAN++ Demo"
|
59 |
+
article = "<div style='text-align: center;'><a href='https://github.com/genforce/interfacegan' target='_blank'>Original Paper</a> | <a href='https://github.com/younesbelkada/interfacegan' target='_blank'>Extended Version</a></div>"
|