Update app.py
Browse files
app.py
CHANGED
@@ -26,11 +26,11 @@ pipe.reshape( batch_size=-1, height=512, width=512, num_images_per_prompt=1)
|
|
26 |
pipe.compile()
|
27 |
|
28 |
|
29 |
-
def infer(prompt,negative_prompt):
|
30 |
|
31 |
image = pipe(
|
32 |
-
prompt = prompt
|
33 |
-
negative_prompt =
|
34 |
width = 512,
|
35 |
height = 512,
|
36 |
guidance_scale=1.0,
|
@@ -84,7 +84,7 @@ with gr.Blocks(css=css) as demo:
|
|
84 |
|
85 |
run_button.click(
|
86 |
fn = infer,
|
87 |
-
inputs = [prompt],
|
88 |
outputs = [result]
|
89 |
)
|
90 |
|
|
|
26 |
pipe.compile()
|
27 |
|
28 |
|
29 |
+
def infer(prompt,negative_prompt, step):
|
30 |
|
31 |
image = pipe(
|
32 |
+
prompt = prompt,
|
33 |
+
negative_prompt = negative_prompt,
|
34 |
width = 512,
|
35 |
height = 512,
|
36 |
guidance_scale=1.0,
|
|
|
84 |
|
85 |
run_button.click(
|
86 |
fn = infer,
|
87 |
+
inputs = [prompt,"",4],
|
88 |
outputs = [result]
|
89 |
)
|
90 |
|