fix customize_embeddings.py

Signed-off-by: Mystic-Slice <vaashwath@gmail.com>

Author: Mystic-Slice

customize/customize_embeddings.py

@@ -33,7 +33,7 @@
 
 # Sentence transformer model HF
 model_path = 'models/all-MiniLM-L6-v2'
-model_id = model_path.split("/")[1]
+model_id = model_path.split("/")[1].lower()
 
 # INPUT FILE
 # Default file with empty embeddings
@@ -43,13 +43,15 @@
 # OUTPUT FILE
 json_out_file_name = f'{json_in_file_name}-{model_id}.json'
 
+print(f'Loading existing data')
+existing_data = None
 # check if the output file already exists
 if os.path.exists(json_out_file_name):
     try:
         # Load existing data from the output file
         existing_data = customize_helper.load_json(json_out_file_name)
     except Exception as e:
-        existing_data = None
+        print(f"Error loading existing data: {e}")
 
 # hashmap 
 prompts_embeddings = {}
@@ -62,8 +64,10 @@
             prompts_embeddings[p["text"]] = p["embedding"]
 
 
-
+print("Populating embeddings and centroids")
 prompt_json = json.load(open(json_in_file))
 prompt_json_embeddings = customize_helper.populate_embeddings(prompt_json, model_path, prompts_embeddings)
 prompt_json_centroids = customize_helper.populate_centroids(prompt_json_embeddings)
+
+print("Saving the embeddings")
 customize_helper.save_json(prompt_json_centroids, json_out_file_name)