File size: 1,289 Bytes
66b0cef
9b4118a
66b0cef
434846f
8a42a65
5901d14
9b4118a
 
 
 
bce0716
9b4118a
5901d14
 
 
 
 
 
 
bce0716
 
147944d
4cb2cd7
1f723fb
4cb2cd7
1f723fb
147944d
bce0716
434846f
f7420b8
9b4118a
09fb843
5901d14
 
9b4118a
f7420b8
 
94f65a5
fd2ee2d
434846f
8a42a65
874c1d4
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
import gradio as gr
import random

model = gr.load("models/Purz/face-projection")

def generate_image(text, seed, style):
    if seed is not None:
        random.seed(seed)

    if text in [example[0] for example in examples]:
        print(f"Using example: {text}")

    if style == "Anime":
        text = f"Anime style, {text}"
    elif style == "Realistic":
        text = f"Realistic style, {text}"
    elif style == "Photographic":
        text = f"Photographic style, {text}"

    return model(text)

examples = [
    ["Humanoid Cat Warrior, Full View", None],
    ["Warhammer Sisterhood", None],
    ["Future Robots war", None],
    ["Fantasy dragon", None]
]

interface = gr.Interface(
    fn=generate_image,
    inputs=[
        gr.Textbox(label="Type here your imagination:", placeholder="Type or click an example..."),
        gr.Slider(minimum=0, maximum=10000, step=1, label="Seed (optional)"),
        gr.Radio(choices=["Anime", "Realistic", "Photographic"], label="Style", value="Realistic")
    ],
    outputs=gr.Image(label="Generated Image"),
    examples=examples,
    theme="NoCrypt/miku",
    description="Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.",
)

interface.launch()