Update app.py
Browse files
app.py
CHANGED
@@ -53,10 +53,9 @@ def generate_image(prompt, negative_prompt, seed, randomize_seed, width, height,
|
|
53 |
height=height,
|
54 |
guidance_scale=guidance_scale,
|
55 |
num_inference_steps=num_inference_steps,
|
56 |
-
num_images_per_prompt=num_images
|
57 |
-
|
58 |
-
|
59 |
-
).images
|
60 |
|
61 |
end_time = time.time()
|
62 |
generation_time = end_time - start_time
|
|
|
53 |
height=height,
|
54 |
guidance_scale=guidance_scale,
|
55 |
num_inference_steps=num_inference_steps,
|
56 |
+
num_images_per_prompt=num_images
|
57 |
+
generator=generator
|
58 |
+
).images
|
|
|
59 |
|
60 |
end_time = time.time()
|
61 |
generation_time = end_time - start_time
|