From d050e1ac88b812bfbc8cdc22d1178f7b7c71e942 Mon Sep 17 00:00:00 2001
From: unknown <alexyalynski@gmail.com>
Date: Thu, 13 Feb 2025 00:13:05 +0200
Subject: [PATCH] fix

---
 color_transfer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/color_transfer.py b/color_transfer.py
index 14f4185..2f4b48e 100644
--- a/color_transfer.py
+++ b/color_transfer.py
@@ -29,7 +29,7 @@ def SwitchColors(image, detected_colors, target_colors, clustering_model, distan
     distance_methods = {
     "Euclidean": EuclideanDistance,
     "Manhattan": ManhattanDistance,
-    "HSV_Similarity": CosineSimilarity
+    "Cosine Similarity": CosineSimilarity
     }
 
     distance_method = distance_methods.get(distance_method)
@@ -55,7 +55,7 @@ class PaletteTransferNode:
                 "target_colors": ("COLOR_LIST",),
                 "color_space": (["RGB", "HSV", "LAB"], {'default': 'RGB'}),
                 "cluster_method": (["Kmeans","Mini batch Kmeans"], {'default': 'Kmeans'}, ),
-                "distance_method": (["Euclidean", "Manhattan", "CosineSimilarity"], {'default': 'Euclidean'}, )
+                "distance_method": (["Euclidean", "Manhattan", "Cosine Similarity"], {'default': 'Euclidean'}, )
                 }
             }
         return data_in