jgrosjean commited on
Commit
8d1c4fd
1 Parent(s): 7524bd5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -35,7 +35,7 @@ import torch
35
  from transformers import AutoModel, AutoTokenizer
36
 
37
  # Load swissBERT for sentence embeddings model
38
- model_name = "jgrosjean-mathesis/swissbert-for-sentence-embeddings"
39
  model = AutoModel.from_pretrained(model_name)
40
  tokenizer = AutoTokenizer.from_pretrained(model_name)
41
 
@@ -161,7 +161,7 @@ Note: For French, Italian and Romansh, the training data remains in German, whil
161
 
162
  ### Results
163
 
164
- Sentence SwissBERT achieves comparable results as the best-performing multilingual Sentence-BERT model (distiluse-base-multilingual-cased) and always outperforms it for German and Romansh.
165
 
166
  | Evaluation task |Swissbert | |Sentence Swissbert| |Sentence-BERT| |
167
  |------------------------|----------|-----------|------------------|-----------|-------------|-----------|
 
35
  from transformers import AutoModel, AutoTokenizer
36
 
37
  # Load swissBERT for sentence embeddings model
38
+ model_name = "jgrosjean-mathesis/sentence-swissbert"
39
  model = AutoModel.from_pretrained(model_name)
40
  tokenizer = AutoTokenizer.from_pretrained(model_name)
41
 
 
161
 
162
  ### Results
163
 
164
+ Sentence SwissBERT achieves comparable results as the best-performing multilingual Sentence-BERT model (distiluse-base-multilingual-cased) and outperforms it for German and Romansh.
165
 
166
  | Evaluation task |Swissbert | |Sentence Swissbert| |Sentence-BERT| |
167
  |------------------------|----------|-----------|------------------|-----------|-------------|-----------|