JoBeer commited on
Commit
848a0c9
1 Parent(s): e24f8b0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -8
README.md CHANGED
@@ -5,6 +5,8 @@ tags:
5
  - feature-extraction
6
  - sentence-similarity
7
  - transformers
 
 
8
  ---
9
 
10
  # mboth/distil-eng-quora-sentence
@@ -27,7 +29,7 @@ Then you can use the model like this:
27
  from sentence_transformers import SentenceTransformer
28
  sentences = ["This is an example sentence", "Each sentence is converted"]
29
 
30
- model = SentenceTransformer('mboth/distil-eng-quora-sentence')
31
  embeddings = model.encode(sentences)
32
  print(embeddings)
33
  ```
@@ -53,8 +55,8 @@ def mean_pooling(model_output, attention_mask):
53
  sentences = ['This is an example sentence', 'Each sentence is converted']
54
 
55
  # Load model from HuggingFace Hub
56
- tokenizer = AutoTokenizer.from_pretrained('mboth/distil-eng-quora-sentence')
57
- model = AutoModel.from_pretrained('mboth/distil-eng-quora-sentence')
58
 
59
  # Tokenize sentences
60
  encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
@@ -86,8 +88,4 @@ SentenceTransformer(
86
  (0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: DistilBertModel
87
  (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False})
88
  )
89
- ```
90
-
91
- ## Citing & Authors
92
-
93
- <!--- Describe where people can find more information -->
 
5
  - feature-extraction
6
  - sentence-similarity
7
  - transformers
8
+ language:
9
+ - en
10
  ---
11
 
12
  # mboth/distil-eng-quora-sentence
 
29
  from sentence_transformers import SentenceTransformer
30
  sentences = ["This is an example sentence", "Each sentence is converted"]
31
 
32
+ model = SentenceTransformer('gart-labor/eng-distilBERT-se')
33
  embeddings = model.encode(sentences)
34
  print(embeddings)
35
  ```
 
55
  sentences = ['This is an example sentence', 'Each sentence is converted']
56
 
57
  # Load model from HuggingFace Hub
58
+ tokenizer = AutoTokenizer.from_pretrained('gart-labor/eng-distilBERT-se')
59
+ model = AutoModel.from_pretrained('gart-labor/eng-distilBERT-se')
60
 
61
  # Tokenize sentences
62
  encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
 
88
  (0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: DistilBertModel
89
  (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False})
90
  )
91
+ ```