Update app.py
Browse files
app.py
CHANGED
|
@@ -63,7 +63,7 @@ def generate_image(prompt: str, negative_prompt: str = "", inference_steps: int
|
|
| 63 |
|
| 64 |
images = images.reshape((num_samples,) + images.shape[-3:])
|
| 65 |
images = pipeline.numpy_to_pil(images)
|
| 66 |
-
return images
|
| 67 |
|
| 68 |
examples = [
|
| 69 |
["A watercolor painting of a bird"],
|
|
|
|
| 63 |
|
| 64 |
images = images.reshape((num_samples,) + images.shape[-3:])
|
| 65 |
images = pipeline.numpy_to_pil(images)
|
| 66 |
+
return images[0]
|
| 67 |
|
| 68 |
examples = [
|
| 69 |
["A watercolor painting of a bird"],
|