Amir Zait commited on
Commit
b5cd199
1 Parent(s): 8ee95b0
Files changed (1) hide show
  1. app.py +1 -1
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 = model.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))
 
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))