Update app.py
Browse files
app.py
CHANGED
@@ -25,15 +25,15 @@ pipe.compile()
|
|
25 |
|
26 |
prompt = ""
|
27 |
negative_prompt = "EasyNegative, "
|
28 |
-
|
29 |
-
def infer(prompt, negative_prompt):
|
30 |
image = pipe(
|
31 |
prompt=prompt,
|
32 |
negative_prompt=negative_prompt,
|
33 |
width=WIDTH,
|
34 |
height=HIGH,
|
35 |
guidance_scale=1.0,
|
36 |
-
num_inference_steps=
|
37 |
num_images_per_prompt=1,
|
38 |
).images[0]
|
39 |
|
|
|
25 |
|
26 |
prompt = ""
|
27 |
negative_prompt = "EasyNegative, "
|
28 |
+
num_inference_steps = 4
|
29 |
+
def infer(prompt, negative_prompt, num_inference_steps):
|
30 |
image = pipe(
|
31 |
prompt=prompt,
|
32 |
negative_prompt=negative_prompt,
|
33 |
width=WIDTH,
|
34 |
height=HIGH,
|
35 |
guidance_scale=1.0,
|
36 |
+
num_inference_steps=num_inference_steps,
|
37 |
num_images_per_prompt=1,
|
38 |
).images[0]
|
39 |
|