use rgb instead of bgr

app.py

@@ -16,7 +16,7 @@ img_path = 'input.png'
 def color(image):
     output = img_colorization(image[...,::-1])
     result = output[OutputKeys.OUTPUT_IMG].astype(np.uint8)
-    result = result[...,::-1]
+    # result = result[...,::-1]
     # Generate a unique filename using UUID
     unique_imgfilename = str(uuid.uuid4()) + '.png'
     cv2.imwrite(unique_imgfilename, result)