Update app.py

app.py

@@ -299,7 +299,7 @@ def visualize(model, data, dims):
     L = model.L
     fake_imgs = lab_to_rgb(L, fake_color)
     real_imgs = lab_to_rgb(L, real_color)
-    img=cv.resize(fake_imgs[0], dsize=(dims[1], dims[0]), interpolation=cv2.INTER_CUBIC)
+    img=cv.resize(fake_imgs[0], dsize=(dims[1], dims[0]), interpolation=cv.INTER_CUBIC)
     return img
     return fake_imgs[0]
     for i in range(1):