Create app.py

app.py

@@ -0,0 +1,64 @@
+import os
+import uuid
+import cv2
+import numpy as np
+import gradio as gr
+from modelscope.pipelines import pipeline
+from modelscope.utils.constant import Tasks
+from modelscope.outputs import OutputKeys
+from gradio_imageslider import ImageSlider
+
+# Initialize the image colorization pipeline with a specified model revision
+img_colorization = pipeline(
+    task=Tasks.image_colorization,
+    model='iic/cv_ddcolor_image-colorization',
+    model_revision='v1.02'  # Specify the model revision to avoid warnings
+)
+
+def colorize_image(input_image):
+    """
+    Colorizes a grayscale image using the DDColor model.
+
+    Args:
+        input_image (numpy.ndarray): Grayscale input image.
+
+    Returns:
+        tuple: Original and colorized images.
+    """
+    # Convert RGB to BGR for OpenCV compatibility
+    bgr_image = cv2.cvtColor(input_image, cv2.COLOR_RGB2BGR)
+
+    # Perform colorization
+    output = img_colorization(bgr_image)
+    colorized_bgr = output[OutputKeys.OUTPUT_IMG].astype(np.uint8)
+
+    # Convert BGR back to RGB for display
+    colorized_rgb = cv2.cvtColor(colorized_bgr, cv2.COLOR_BGR2RGB)
+
+    return input_image, colorized_rgb
+
+# Define the Gradio interface
+title = "DDColor: Photo-Realistic Image Colorization"
+description = (
+    "Upload a black & white image to colorize it using the DDColor model. "
+    "This application utilizes a dual-decoder architecture to produce vivid and natural colorizations."
+)
+
+# Example images (ensure these paths are correct or adjust accordingly)
+examples = [['./input.jpg']]
+
+# Create the Gradio interface
+demo = gr.Interface(
+    fn=colorize_image,
+    inputs=gr.Image(label="Upload Grayscale Image"),
+    outputs=ImageSlider(position=0.5, label='Colorization Comparison'),
+    examples=examples,
+    title=title,
+    description=description,
+    allow_flagging="never"
+)
+
+if __name__ == "__main__":
+    # Launch the Gradio app with production settings
+    demo.launch(server_name="0.0.0.0", server_port=7860, share=False)
+