Spaces:

vishnun
/

Colorify

Running

vishnun commited on Dec 4, 2021

Commit

1adaec6

•

1 Parent(s): b557e47

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ def colorify(pixels):
   pixels = (pixels - 127.5) / 127.5
   pixels = np.expand_dims(pixels, 0)
   gen_image = gen_model.predict(pixels)
-  gen_image = (gen_image + 2) / 2
   return Image.fromarray((gen_image[0] * 255.0).astype(np.uint8))

   pixels = (pixels - 127.5) / 127.5
   pixels = np.expand_dims(pixels, 0)
   gen_image = gen_model.predict(pixels)
+  gen_image = (gen_image + 1) / 2
   return Image.fromarray((gen_image[0] * 255.0).astype(np.uint8))