hooman650 commited on
Commit
7c03ea6
1 Parent(s): aa0b863

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -8,6 +8,8 @@ library_name: transformers
8
 
9
  O4 optimized weights of [`NeuML/pubmedbert-base-embeddings-matryoshka`](https://huggingface.co/NeuML/pubmedbert-base-embeddings-matryoshka).
10
 
 
 
11
  ```python
12
  from optimum.onnxruntime import ORTModelForFeatureExtraction
13
  from transformers import AutoTokenizer
@@ -39,4 +41,6 @@ embeddings = meanpooling(output, inputs['attention_mask'])
39
  dimensions = 256
40
 
41
  print("Sentence embeddings:")
42
- print(embeddings[:, :dimensions])
 
 
 
8
 
9
  O4 optimized weights of [`NeuML/pubmedbert-base-embeddings-matryoshka`](https://huggingface.co/NeuML/pubmedbert-base-embeddings-matryoshka).
10
 
11
+ ## Usage
12
+
13
  ```python
14
  from optimum.onnxruntime import ORTModelForFeatureExtraction
15
  from transformers import AutoTokenizer
 
41
  dimensions = 256
42
 
43
  print("Sentence embeddings:")
44
+ print(embeddings[:, :dimensions])
45
+
46
+ ```