diff --git a/aimodel/src/lib/vis/embeddings.py b/aimodel/src/lib/vis/embeddings.py
index a0698b7..0df6aa2 100644
--- a/aimodel/src/lib/vis/embeddings.py
+++ b/aimodel/src/lib/vis/embeddings.py
@@ -42,4 +42,5 @@ def vis_embeddings(filepath_output, features):
 	axes["B"].set_title(f"Parallel coordinates plot", fontsize=20)
 	
 	plt.suptitle(f"RainfallContrastive embeddings | rainfall | E2 ConvNeXt | {len(features)} items", fontsize=28, weight="bold")
-	plt.savefig(filepath_output)
\ No newline at end of file
+	plt.savefig(filepath_output)
+	plt.close()
\ No newline at end of file