Update app.py

app.py

@@ -299,8 +299,8 @@ 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=cv.INTER_CUBIC)
-    return img
+    # img=cv.resize(fake_imgs[0], dsize=(dims[1], dims[0]), interpolation=cv.INTER_CUBIC)
+    return fake_imgs[0].resize((dims[1], dims[0]))
     return fake_imgs[0]
     for i in range(1):
         # t_img = transforms.Resize((dims[0], dims[1]))(t_img)
@@ -375,7 +375,7 @@ description = "Gradio demo for Image Colorization project. You can give an image
 gr.Interface(
     main_func, 
     [gr.Image(type="filepath", label="Input Image") ], 
-    [gr.Textbox(label="Image Size"),gr.Image(type="numpy", label="Output Image")],
+    [gr.Textbox(label="Image Size"),gr.Image(type="pil", label="Output Image")],
     title=title,
     description=description,
     # examples=[['a.png', ['Generate text']], ['b.png', ['Generate text','Summary','Important Sentences']], ]