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 |
|