bugfix

app.py

@@ -53,7 +53,7 @@ def generate_image(text):
         condition_scale=cond_scale,
         )
     encoded_images = encoded_images.sequences[..., 1:]
-    decoded_images = vqgan.decode(encoded_images, vqgan.params)
+    decoded_images = vqgan.decode_code(encoded_images, vqgan.params)
     decoded_images = decoded_images.clip(0.0, 1.0).reshape((-1, 256, 256, 3))
     img = decoded_images[0]
     return Image.fromarray(np.asarray(img * 255, dtype=np.uint8))