Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ batch_size, num_images, height, width = 1, 1, 512, 512
|
|
18 |
pipeline.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
|
19 |
pipeline.compile()
|
20 |
|
21 |
-
def infer(prompt, num_inference_steps):
|
22 |
|
23 |
image = pipeline(
|
24 |
prompt = prompt,
|
@@ -93,7 +93,7 @@ with gr.Blocks(css=css) as demo:
|
|
93 |
|
94 |
run_button.click(
|
95 |
fn = infer,
|
96 |
-
inputs = [prompt, num_inference_steps],
|
97 |
outputs = [result]
|
98 |
)
|
99 |
|
|
|
18 |
pipeline.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
|
19 |
pipeline.compile()
|
20 |
|
21 |
+
def infer(prompt,negative_prompt, num_inference_steps):
|
22 |
|
23 |
image = pipeline(
|
24 |
prompt = prompt,
|
|
|
93 |
|
94 |
run_button.click(
|
95 |
fn = infer,
|
96 |
+
inputs = [prompt, negative_prompt,num_inference_steps],
|
97 |
outputs = [result]
|
98 |
)
|
99 |
|