vishnun commited on
Commit
d4a0e78
1 Parent(s): ba39732

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def colorify(pixels):
16
  gen_image = gen_model.predict(pixels)
17
  gen_image = (gen_image + 1) / 1.5
18
 
19
- return Image.fromarray(gen_image[0]*255)
20
 
21
  title = "Colorify"
22
  description = "Recolor your images using this lite version of PIX2PIX GAN"
 
16
  gen_image = gen_model.predict(pixels)
17
  gen_image = (gen_image + 1) / 1.5
18
 
19
+ return gen_image[0]*255
20
 
21
  title = "Colorify"
22
  description = "Recolor your images using this lite version of PIX2PIX GAN"