Commit
·
711e084
1
Parent(s):
bfddad0
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ def generate_image(prompt):
|
|
| 11 |
num_inference_steps = 20
|
| 12 |
|
| 13 |
generator = torch.Generator("cuda").manual_seed(0)
|
| 14 |
-
image = pipeline(prompt, generator=generator,
|
| 15 |
|
| 16 |
return image
|
| 17 |
|
|
|
|
| 11 |
num_inference_steps = 20
|
| 12 |
|
| 13 |
generator = torch.Generator("cuda").manual_seed(0)
|
| 14 |
+
image = pipeline(prompt, generator=generator, num_inference_steps=20).images[0]
|
| 15 |
|
| 16 |
return image
|
| 17 |
|