File size: 1,376 Bytes
f1ac645
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
import gradio as gr

def generate_image(prompt, negative_prompt, seed, width, height, prior_inference_steps, prior_guidance_scale, decoder_inference_steps, decoder_guidance_scale, num_images):
    client = gr.Client("multimodalart/stable-cascade")
    result = client.predict(
        prompt,
        negative_prompt,
        seed,
        width,
        height,
        prior_inference_steps,
        prior_guidance_scale,
        decoder_inference_steps,
        decoder_guidance_scale,
        num_images,
        api_name="/run"
    )
    return result

demo = gr.Interface(
    fn=generate_image,
    inputs=[
        gr.Textbox(label="Prompt"),
        gr.Textbox(label="Negative prompt"),
        gr.Slider(label="Seed", minimum=0, maximum=2147483647, step=1),
        gr.Slider(label="Width", minimum=1024, maximum=1536, step=1),
        gr.Slider(label="Height", minimum=1024, maximum=1536, step=1),
        gr.Slider(label="Prior Inference Steps", minimum=10, maximum=30, step=1),
        gr.Slider(label="Prior Guidance Scale", minimum=0, maximum=20, step=1),
        gr.Slider(label="Decoder Inference Steps", minimum=4, maximum=12, step=1),
        gr.Slider(label="Decoder Guidance Scale", minimum=0, maximum=0, step=1),
        gr.Slider(label="Number of Images", minimum=1, maximum=2, step=1),
    ],
    outputs=["image"],
    theme = "soft"
)

demo.launch()