diff --git a/NN_server/Models/ensemble_1200_v44.py b/NN_server/Models/ensemble_1200_v44.py
index bc285e2..49d2548 100644
--- a/NN_server/Models/ensemble_1200_v44.py
+++ b/NN_server/Models/ensemble_1200_v44.py
@@ -244,7 +244,7 @@ def post_func_ensemble(src="", model_type="", prediction="", model_id=0, ind_inf
 
         if isinstance(data, (list, tuple)) and len(data) >= 2:
             fig, ax = plt.subplots()
-            ax.imshow(_as_training_colormap_image(data[0]), cmap="viridis")
+            ax.imshow(_as_training_colormap_image(data[0]), cmap="plasma")
             plt.savefig(src + "_inference_" + str(ind_inference) + "_" + prediction + "_real_" + str(model_id) + "_" + model_type + ".png")
             plt.clf()
             plt.cla()
@@ -253,7 +253,7 @@ def post_func_ensemble(src="", model_type="", prediction="", model_id=0, ind_inf
             gc.collect()
 
             fig, ax = plt.subplots()
-            ax.imshow(_as_training_colormap_image(data[1]), cmap="viridis")
+            ax.imshow(_as_training_colormap_image(data[1]), cmap="plasma")
             plt.savefig(src + "_inference_" + str(ind_inference) + "_" + prediction + "_mod_" + str(model_id) + "_" + model_type + ".png")
             plt.clf()
             plt.cla()
diff --git a/NN_server/Models/ensemble_2400_v44.py b/NN_server/Models/ensemble_2400_v44.py
index bc285e2..49d2548 100644
--- a/NN_server/Models/ensemble_2400_v44.py
+++ b/NN_server/Models/ensemble_2400_v44.py
@@ -244,7 +244,7 @@ def post_func_ensemble(src="", model_type="", prediction="", model_id=0, ind_inf
 
         if isinstance(data, (list, tuple)) and len(data) >= 2:
             fig, ax = plt.subplots()
-            ax.imshow(_as_training_colormap_image(data[0]), cmap="viridis")
+            ax.imshow(_as_training_colormap_image(data[0]), cmap="plasma")
             plt.savefig(src + "_inference_" + str(ind_inference) + "_" + prediction + "_real_" + str(model_id) + "_" + model_type + ".png")
             plt.clf()
             plt.cla()
@@ -253,7 +253,7 @@ def post_func_ensemble(src="", model_type="", prediction="", model_id=0, ind_inf
             gc.collect()
 
             fig, ax = plt.subplots()
-            ax.imshow(_as_training_colormap_image(data[1]), cmap="viridis")
+            ax.imshow(_as_training_colormap_image(data[1]), cmap="plasma")
             plt.savefig(src + "_inference_" + str(ind_inference) + "_" + prediction + "_mod_" + str(model_id) + "_" + model_type + ".png")
             plt.clf()
             plt.cla()
diff --git a/NN_server/Models/ensemble_915_v44.py b/NN_server/Models/ensemble_915_v44.py
index 2f3b53b..a6debd7 100644
--- a/NN_server/Models/ensemble_915_v44.py
+++ b/NN_server/Models/ensemble_915_v44.py
@@ -216,7 +216,7 @@ def post_func_ensemble(src="", model_type="", prediction="", model_id=0, ind_inf
 
         if isinstance(data, (list, tuple)) and len(data) >= 2:
             fig, ax = plt.subplots()
-            ax.imshow(_as_training_colormap_image(data[0]), cmap="viridis")
+            ax.imshow(_as_training_colormap_image(data[0]), cmap="plasma")
             plt.savefig(src + "_inference_" + str(ind_inference) + "_" + prediction + "_real_" + str(model_id) + "_" + model_type + ".png")
             plt.clf()
             plt.cla()
@@ -225,7 +225,7 @@ def post_func_ensemble(src="", model_type="", prediction="", model_id=0, ind_inf
             gc.collect()
 
             fig, ax = plt.subplots()
-            ax.imshow(_as_training_colormap_image(data[1]), cmap="viridis")
+            ax.imshow(_as_training_colormap_image(data[1]), cmap="plasma")
             plt.savefig(src + "_inference_" + str(ind_inference) + "_" + prediction + "_mod_" + str(model_id) + "_" + model_type + ".png")
             plt.clf()
             plt.cla()