Update app.py

app.py

@@ -299,12 +299,12 @@ 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(np.uint8(fake_imgs[0]), dsize=(dims[1], dims[0]), interpolation=cv2.INTER_CUBIC)
-    return img
+    # img=cv.resize(np.uint8(fake_imgs[0]), dsize=(dims[1], dims[0]), interpolation=cv2.INTER_CUBIC)
+    # return img
     for i in range(1):
         # t_img = transforms.Resize((dims[0], dims[1]))(t_img)
         img = Image.fromarray(np.uint8(fake_imgs[i]))
-        img = cv.resize(fake_imgs[i], dsize=(dims[1], dims[0]), interpolation=cv.INTER_CUBIC)
+        img = np.uint8(cv.resize(fake_imgs[i], dsize=(dims[1], dims[0]), interpolation=cv.INTER_CUBIC))
         return img
         # st.text(f"Size of fake image {fake_imgs[i].shape} \n Type of image = {type(fake_imgs[i])}")
         # st.image(img, caption="Output image", use_column_width='auto', clamp=True)
@@ -373,8 +373,8 @@ title = "Image Colorization"
 description = "Gradio demo for Image Colorization project. You can give an image as input on the left side and then click on the submit button. The program would recolorize the image"
 gr.Interface(
     main_func, 
-    [gr.inputs.Image(type="filepath", label="Input Image") ], 
-    [gr.outputs.Textbox(label="Image Size"),gr.outputs.Image(type="numpy", label="Output Image")],
+    [grImage(type="filepath", label="Input Image") ], 
+    [gr.Textbox(label="Image Size"),gr.Image(type="numpy", label="Output Image")],
     title=title,
     description=description,
     # examples=[['a.png', ['Generate text']], ['b.png', ['Generate text','Summary','Important Sentences']], ]