Update app.py

app.py

@@ -372,7 +372,7 @@ description = "Gradio demo for Image Colorization project. You can give an image
 gr.Interface(
     main_func, 
     [gr.inputs.Image(type="filepath", label="Input Image") ], 
-    [gr.outputs.Textbox(label="Image Size"),"image"],
+    [gr.outputs.Textbox(label="Image Size"),gr.outputs.Image(type="pil", label="Output Image")],
     title=title,
     description=description,
     # examples=[['a.png', ['Generate text']], ['b.png', ['Generate text','Summary','Important Sentences']], ]