Hugging Face's logo

Spaces:
Hierarchy-Transformers
/
README
Running

App Files Community
lawhy commited on
Commit
37ac8dd
·
verified ·
1 Parent(s): 6f6e1be

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -23,7 +23,7 @@ gpu_id = 0
23
  device = get_torch_device(gpu_id)
24
 
25
  # load the model
26
- model = HierarchyTransformer.load_pretrained('Hierarchy-Transformers/HiT-MiniLM-L12-WordNet')
27
 
28
  # entity names to be encoded.
29
  entity_names = ["computer", "personal computer", "fruit", "berry"]
 
23
  device = get_torch_device(gpu_id)
24
 
25
  # load the model
26
+ model = HierarchyTransformer.load_pretrained('Hierarchy-Transformers/HiT-MiniLM-L12-WordNet', device)
27
 
28
  # entity names to be encoded.
29
  entity_names = ["computer", "personal computer", "fruit", "berry"]