Kvikontent commited on
Commit
f1ac645
·
verified ·
1 Parent(s): dbbb96b

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +38 -0
app.py ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+
3
+ def generate_image(prompt, negative_prompt, seed, width, height, prior_inference_steps, prior_guidance_scale, decoder_inference_steps, decoder_guidance_scale, num_images):
4
+ client = gr.Client("multimodalart/stable-cascade")
5
+ result = client.predict(
6
+ prompt,
7
+ negative_prompt,
8
+ seed,
9
+ width,
10
+ height,
11
+ prior_inference_steps,
12
+ prior_guidance_scale,
13
+ decoder_inference_steps,
14
+ decoder_guidance_scale,
15
+ num_images,
16
+ api_name="/run"
17
+ )
18
+ return result
19
+
20
+ demo = gr.Interface(
21
+ fn=generate_image,
22
+ inputs=[
23
+ gr.Textbox(label="Prompt"),
24
+ gr.Textbox(label="Negative prompt"),
25
+ gr.Slider(label="Seed", minimum=0, maximum=2147483647, step=1),
26
+ gr.Slider(label="Width", minimum=1024, maximum=1536, step=1),
27
+ gr.Slider(label="Height", minimum=1024, maximum=1536, step=1),
28
+ gr.Slider(label="Prior Inference Steps", minimum=10, maximum=30, step=1),
29
+ gr.Slider(label="Prior Guidance Scale", minimum=0, maximum=20, step=1),
30
+ gr.Slider(label="Decoder Inference Steps", minimum=4, maximum=12, step=1),
31
+ gr.Slider(label="Decoder Guidance Scale", minimum=0, maximum=0, step=1),
32
+ gr.Slider(label="Number of Images", minimum=1, maximum=2, step=1),
33
+ ],
34
+ outputs=["image"],
35
+ theme = "soft"
36
+ )
37
+
38
+ demo.launch()