File size: 1,046 Bytes
006354e
 
e0f92a0
006354e
2783b1f
e0f92a0
 
 
 
 
006354e
 
 
 
 
 
e0f92a0
 
006354e
e0f92a0
 
 
 
 
 
 
 
006354e
e0f92a0
 
 
 
 
 
 
 
006354e
e0f92a0
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import gradio as gr


def set_preset(config_str):
    print(config_str)
    choices = [
        "Small Masked Changes (e.g. add lipstick)",
        "Major Masked Changes (e.g. change hair color or nose size)",
        "Major Global Changes (e.g. change race / gender",
    ]
    if config_str == choices[0]:
        return set_small_local()
    elif config_str == choices[1]:
        return set_major_local()
    elif config_str == choices[2]:
        return set_major_global()


def set_small_local():
    return (
        gr.Slider.update(value=25),
        gr.Slider.update(value=0.15),
        gr.Slider.update(value=1),
        gr.Slider.update(value=4),
    )


def set_major_local():
    return (
        gr.Slider.update(value=25),
        gr.Slider.update(value=0.25),
        gr.Slider.update(value=35),
        gr.Slider.update(value=10),
    )


def set_major_global():
    return (
        gr.Slider.update(value=30),
        gr.Slider.update(value=0.1),
        gr.Slider.update(value=2),
        gr.Slider.update(value=0.2),
    )