Spaces:

vishnun
/

Colorify

Runtime error

vishnun commited on Dec 4, 2021

Commit

a58e0d6

•

1 Parent(s): 6c4aa25

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ def colorify(pixels):
   gen_image = gen_model.predict(pixels)
   gen_image = (gen_image + 1) / 1.5
-  return gen_image[0]
 title = "Colorify"
 description = "Recolor your images using this lite version of PIX2PIX GAN"

   gen_image = gen_model.predict(pixels)
   gen_image = (gen_image + 1) / 1.5
+  return Image.fromarray(gen_image[0]*255)
 title = "Colorify"
 description = "Recolor your images using this lite version of PIX2PIX GAN"