Update colorization.py

colorization.py

@@ -19,7 +19,8 @@ def colorize_image(image):
     # if image is not None:
     #     image.save(f"./output/{now}-input.jpg")    
     model.predict(image.name)
-    return './output/DeOldify/'+Path(image.name).stem+".png"
+    
+    return f'./output/DeOldify/'+Path(image.name).stem+".png"
 
 
 def create_interface():
@@ -28,8 +29,8 @@ def create_interface():
         with gr.Column(scale=1, label = "Colorize photo", visible=True) as colorize_column:
             colorize_input = gr.Image(type="file")
             colorize_button = gr.Button("Colorize!")
-            colorize_output = gr.Image(type="file")
-            download_colorize_button = gr.outputs.File(label="Download colorized image!")
+            colorize_output = gr.outputs.Image(type="numpy", label="Output (The Colorized image)")
+            download_colorize_button = gr.outputs.File(label="Download the colorized image!")
             colorize_button.click(colorize_image, inputs=colorize_input, outputs=colorize_output)
     enhancer.launch()