Spaces:
Runtime error
Runtime error
Amir Zait
commited on
Commit
•
b5cd199
1
Parent(s):
8ee95b0
bugfix
Browse files
app.py
CHANGED
@@ -53,7 +53,7 @@ def generate_image(text):
|
|
53 |
condition_scale=cond_scale,
|
54 |
)
|
55 |
encoded_images = encoded_images.sequences[..., 1:]
|
56 |
-
decoded_images =
|
57 |
decoded_images = decoded_images.clip(0.0, 1.0).reshape((-1, 256, 256, 3))
|
58 |
img = decoded_images[0]
|
59 |
return Image.fromarray(np.asarray(img * 255, dtype=np.uint8))
|
|
|
53 |
condition_scale=cond_scale,
|
54 |
)
|
55 |
encoded_images = encoded_images.sequences[..., 1:]
|
56 |
+
decoded_images = vqgan.decode(encoded_images, vqgan.params)
|
57 |
decoded_images = decoded_images.clip(0.0, 1.0).reshape((-1, 256, 256, 3))
|
58 |
img = decoded_images[0]
|
59 |
return Image.fromarray(np.asarray(img * 255, dtype=np.uint8))
|