Update app.py

app.py

@@ -4,7 +4,7 @@ import colorizers as c
 
 from colorizers.util import postprocess_tens, preprocess_img
 
-def interface(image, model: str):
+def interface(image: Image, model: str):
     if model == "eccv16":
         img = c.eccv16(pretrained=True).eval()
     else:
@@ -27,5 +27,8 @@ gr.Interface(
         gr.inputs.Radio([
             "eccv16", "siggraph17"
         ], type="value", default="eccv16", label="Model")
+    ],
+    [
+        gr.outputs.Image(label="Output")
     ]
 )