Update app.py
Browse files
app.py
CHANGED
@@ -50,7 +50,7 @@ def infer(prompt, num_inference_steps):
|
|
50 |
num_inference_steps = num_inference_steps,
|
51 |
width = 512,
|
52 |
height = 512,
|
53 |
-
num_images_per_prompt=
|
54 |
).images[0]
|
55 |
|
56 |
return image
|
|
|
50 |
num_inference_steps = num_inference_steps,
|
51 |
width = 512,
|
52 |
height = 512,
|
53 |
+
num_images_per_prompt=1,
|
54 |
).images[0]
|
55 |
|
56 |
return image
|