Commit ·
0ca6778
1
Parent(s): b4ec43b
Update README.md
Browse files
README.md
CHANGED
|
@@ -49,11 +49,15 @@ def mean_pooling(model_output, attention_mask):
|
|
| 49 |
|
| 50 |
|
| 51 |
# Sentences we want sentence embeddings for
|
| 52 |
-
sentences = [
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
|
| 54 |
# Load model from HuggingFace Hub
|
| 55 |
tokenizer = AutoTokenizer.from_pretrained('sentence-transformers/all-mpnet-base-v2')
|
| 56 |
-
model = AutoModel.from_pretrained('
|
| 57 |
|
| 58 |
# Tokenize sentences
|
| 59 |
encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
|
|
|
|
| 49 |
|
| 50 |
|
| 51 |
# Sentences we want sentence embeddings for
|
| 52 |
+
sentences = [
|
| 53 |
+
"Human-Computer Interaction (HCI) can enhance emotion regulation training",
|
| 54 |
+
"One of the most effective emotion regulation in youth is through cognitive reapproval",
|
| 55 |
+
"Business analytics is an extensive form of using data"
|
| 56 |
+
]
|
| 57 |
|
| 58 |
# Load model from HuggingFace Hub
|
| 59 |
tokenizer = AutoTokenizer.from_pretrained('sentence-transformers/all-mpnet-base-v2')
|
| 60 |
+
model = AutoModel.from_pretrained('salsabiilashifa11/sbert-paper')
|
| 61 |
|
| 62 |
# Tokenize sentences
|
| 63 |
encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
|