biu-nlp
/

abstract-sim-query

Feature Extraction

sentence-similarity

Inference Endpoints

Model card Files Files and versions Community

ravfogs commited on May 13, 2023

Commit

afa2917

·

1 Parent(s): b98460f

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/description-transformer-mpnet-wiki-sentence-encoder")
-        query_encoder = AutoModel.from_pretrained("ravfogs/description-transformer-mpnet-wiki-query-encoder")
-        tokenizer = AutoTokenizer.from_pretrained("ravfogs/description-transformer-mpnet-wiki-sentence-encoder")
         return tokenizer, query_encoder, sentence_encoder

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