Update app.py

app.py

@@ -300,7 +300,7 @@ def visualize(model, data, dims):
     fake_imgs = lab_to_rgb(L, fake_color)
     real_imgs = lab_to_rgb(L, real_color)
     # img=cv.resize(np.uint8(fake_imgs[0]), dsize=(dims[1], dims[0]), interpolation=cv2.INTER_CUBIC)
-    return np.uint8(fake_imgs[0])
+    return fake_imgs[0]
     for i in range(1):
         # t_img = transforms.Resize((dims[0], dims[1]))(t_img)
         img = Image.fromarray(np.uint8(fake_imgs[i]))