Update app.py
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ pipe.compile()
|
|
| 28 |
prompt=""
|
| 29 |
negative_prompt=""
|
| 30 |
|
| 31 |
-
def infer(prompt,negative_prompt
|
| 32 |
|
| 33 |
image = pipe(
|
| 34 |
prompt = prompt,
|
|
@@ -36,7 +36,7 @@ def infer(prompt,negative_prompt, num_inference_steps):
|
|
| 36 |
width = 512,
|
| 37 |
height = 512,
|
| 38 |
guidance_scale=1.0,
|
| 39 |
-
num_inference_steps=
|
| 40 |
num_images_per_prompt=1,
|
| 41 |
).images[0]
|
| 42 |
|
|
@@ -86,7 +86,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 86 |
|
| 87 |
run_button.click(
|
| 88 |
fn = infer,
|
| 89 |
-
inputs = [prompt,negative_prompt
|
| 90 |
outputs = [result]
|
| 91 |
)
|
| 92 |
|
|
|
|
| 28 |
prompt=""
|
| 29 |
negative_prompt=""
|
| 30 |
|
| 31 |
+
def infer(prompt,negative_prompt):
|
| 32 |
|
| 33 |
image = pipe(
|
| 34 |
prompt = prompt,
|
|
|
|
| 36 |
width = 512,
|
| 37 |
height = 512,
|
| 38 |
guidance_scale=1.0,
|
| 39 |
+
num_inference_steps=8,
|
| 40 |
num_images_per_prompt=1,
|
| 41 |
).images[0]
|
| 42 |
|
|
|
|
| 86 |
|
| 87 |
run_button.click(
|
| 88 |
fn = infer,
|
| 89 |
+
inputs = [prompt,negative_prompt],
|
| 90 |
outputs = [result]
|
| 91 |
)
|
| 92 |
|