biu-nlp
/

abstract-sim-query-pubmed

Feature Extraction

sentence-similarity

text-embeddings-inference

Model card Files Files and versions

ravfogs commited on May 14, 2023

Commit

a4aa9d3

·

1 Parent(s): 47364be

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -14,9 +14,9 @@ import torch
 def load_finetuned_model():
-        sentence_encoder = AutoModel.from_pretrained("ravfogs/abstract-sim-sentence-pubmed")
-        query_encoder = AutoModel.from_pretrained("ravfogs/abstract-sim-query-pubmed")
-        tokenizer = AutoTokenizer.from_pretrained("ravfogs/abstract-sim-sentence-pubmed")
         return tokenizer, query_encoder, sentence_encoder

 def load_finetuned_model():
+        sentence_encoder = AutoModel.from_pretrained("biu-nlp/abstract-sim-sentence-pubmed")
+        query_encoder = AutoModel.from_pretrained("biu-nlp/abstract-sim-query-pubmed")
+        tokenizer = AutoTokenizer.from_pretrained("biu-nlp/abstract-sim-sentence-pubmed")
         return tokenizer, query_encoder, sentence_encoder