Update app.py

app.py

@@ -301,7 +301,7 @@ def visualize(model, data, dims):
     real_imgs = lab_to_rgb(L, real_color)
     # img=cv.resize(fake_imgs[0], dsize=(dims[1], dims[0]), interpolation=cv.INTER_CUBIC)
     img=fake_imgs[0]
-    # return np.resize(img,(dims[1], dims[0]))
+    return np.resize(img,(dims[1], dims[0]))
     return fake_imgs[0]
     for i in range(1):
         # t_img = transforms.Resize((dims[0], dims[1]))(t_img)
@@ -369,15 +369,14 @@ def main_func(filepath):
         model.setup_input(data)
         model.optimize()
         img=visualize(model, data, a)
-        img=img.resize(a)
-    return (size_text,img)
+    return (size_text,img,img,img)
 
 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.Image(type="filepath", label="Input Image") ], 
-    [gr.Textbox(label="Image Size"),gr.Image(type="pil", label="Output Image")],
+    [gr.Textbox(label="Image Size"),gr.Image(type="pil", label="Output Image"),gr.Image( label="Output Image2"),"image"],
     title=title,
     description=description,
     examples=[ ['face.jpg'], ['montains.jpg'],['tree.jpg']  ]