Hiveurban
/

multilingual-e5-large-pooled

Feature Extraction

sentence-transformers

Sentence Transformers

sentence-similarity

text-embeddings-inference

Model card Files Files and versions

shamaayan commited on Sep 24, 2024

Commit

c7c1092

·

1 Parent(s): 0d1febe

model(data, padding=False, truncation=True)

Files changed (1) hide show

code/inference.py +1 -1

code/inference.py CHANGED Viewed

@@ -3,6 +3,6 @@ from typing import List, Union
 def predict_fn(data: Union[List[str], str], model):
-    outputs = model(input_texts, padding=False, truncation=True)
     embeddings = [np.array(r[0]).mean(axis=0).tolist() for r in outputs]
     return embeddings

 def predict_fn(data: Union[List[str], str], model):
+    outputs = model(data, padding=False, truncation=True)
     embeddings = [np.array(r[0]).mean(axis=0).tolist() for r in outputs]
     return embeddings