obrizum
/

all-mpnet-base-v2

Feature Extraction

sentence-transformers

sentence-similarity

Inference Endpoints

Model card Files Files and versions Community

obrizum commited on May 5, 2022

Commit

414113c

•

1 Parent(s): cd1ebd5

update readme

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -40,8 +40,8 @@ def mean_pooling(model_output, attention_mask):
 sentences = ['This is an example sentence', 'Each sentence is converted']
 # Load model from HuggingFace Hub
-tokenizer = AutoTokenizer.from_pretrained('sentence-transformers/all-mpnet-base-v2')
-model = AutoModel.from_pretrained('sentence-transformers/all-mpnet-base-v2')
 # Tokenize sentences
 encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')

 sentences = ['This is an example sentence', 'Each sentence is converted']
 # Load model from HuggingFace Hub
+tokenizer = AutoTokenizer.from_pretrained('obrizum/all-mpnet-base-v2')
+model = AutoModel.from_pretrained('obrizum/all-mpnet-base-v2')
 # Tokenize sentences
 encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')