Hugging Face's logo

philipp-zettl
/
BAAI-bge-m3-ONNX

Feature Extraction
sentence-transformers
ONNX
xlm-roberta
text-embeddings-inference
Inference Endpoints
Model card Files Community
philipp-zettl commited on
Commit
3524ae6
·
verified ·
1 Parent(s): 21da3e2

add missing import

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -31,7 +31,7 @@ model = ORTModelForFeatureExtraction.from_pretrained("philipp-zettl/BAAI-bge-m3-
31
  ```python
32
  import torch
33
  import torch.nn.functional as F
34
- from typing import Any, Dict
35
 
36
 
37
  def mean_pooling(model_output, attention_mask):
 
31
  ```python
32
  import torch
33
  import torch.nn.functional as F
34
+ from sklearn.metrics.pairwise import cosine_similarity
35
 
36
 
37
  def mean_pooling(model_output, attention_mask):