Spaces:

Hierarchy-Transformers
/

README

Running

lawhy commited on Jan 20, 2024

Commit

55898f4

verified ·

1 Parent(s): f55ed24

Update README.md

Files changed (1) hide show

README.md CHANGED Viewed

@@ -26,7 +26,7 @@ device = get_torch_device(gpu_id)
 model = HierarchyTransformer.load_pretrained('Hierarchy-Transformers/HiT-MiniLM-L12-WordNet')
 # entity names to be encoded.
-entity_names = ["computer", "personal computer", "laptop"]
 # get the entity embeddings
 entity_embeddings = model.encode(entity_names)

 model = HierarchyTransformer.load_pretrained('Hierarchy-Transformers/HiT-MiniLM-L12-WordNet')
 # entity names to be encoded.
+entity_names = ["computer", "personal computer", "fruit", "berry"]
 # get the entity embeddings
 entity_embeddings = model.encode(entity_names)