biu-nlp
/

abstract-sim-sentence-pubmed

Feature Extraction

sentence-similarity

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

ravfogs commited on Oct 1, 2023

Commit

c06fd72

·

1 Parent(s): 3eef34e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -25,7 +25,7 @@ def load_finetuned_model():
         return tokenizer, query_encoder, sentence_encoder
-def encode_batch_fn(model, tokenizer, sentences, device)
     input_ids = tokenizer(sentences, padding=True, max_length=128, truncation=True, return_tensors="pt",
                           add_special_tokens=True).to(device)
     features = model(**input_ids)[0]

         return tokenizer, query_encoder, sentence_encoder
+def encode_batch(model, tokenizer, sentences, device):
     input_ids = tokenizer(sentences, padding=True, max_length=128, truncation=True, return_tensors="pt",
                           add_special_tokens=True).to(device)
     features = model(**input_ids)[0]