Spaces:
Runtime error
Runtime error
Michael-Geis
commited on
Commit
•
c8cfc43
1
Parent(s):
6cc55f5
fixed error in generate_tag_embeddings
Browse files- embedding.py +1 -1
embedding.py
CHANGED
@@ -105,7 +105,7 @@ def generate_tag_embeddings(model_name, path_to_tag_dict, path_to_save_embedding
|
|
105 |
dict_string = file.read()
|
106 |
tag_dict = json.loads(dict_string)
|
107 |
|
108 |
-
tag_name_list = list(tag_dict.values())
|
109 |
embedded_tag_names = model.encode(sentences=tag_name_list, show_progress_bar=True)
|
110 |
embedded_tag_names_df = pd.DataFrame(embedded_tag_names)
|
111 |
embedded_tag_names_df.columns = [
|
|
|
105 |
dict_string = file.read()
|
106 |
tag_dict = json.loads(dict_string)
|
107 |
|
108 |
+
tag_name_list = list(set(tag_dict.values()))
|
109 |
embedded_tag_names = model.encode(sentences=tag_name_list, show_progress_bar=True)
|
110 |
embedded_tag_names_df = pd.DataFrame(embedded_tag_names)
|
111 |
embedded_tag_names_df.columns = [
|