salsabiilashifa11 commited on
Commit
0ca6778
·
1 Parent(s): b4ec43b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -2
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 = ['This is an example sentence', 'Each sentence is converted']
 
 
 
 
53
 
54
  # Load model from HuggingFace Hub
55
  tokenizer = AutoTokenizer.from_pretrained('sentence-transformers/all-mpnet-base-v2')
56
- model = AutoModel.from_pretrained('sentence-transformers/all-mpnet-base-v2')
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')