asier-gutierrez commited on
Commit
625c998
·
1 Parent(s): cdae411

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -1
README.md CHANGED
@@ -3,4 +3,19 @@ language:
3
  - eu
4
  ---
5
 
6
- # Basque Semantic Textual Similarity Model
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  - eu
4
  ---
5
 
6
+ # Basque Semantic Textual Similarity Model
7
+
8
+ ```
9
+ import transformers
10
+
11
+ if __name__ == '__main__':
12
+ model = 'eu-sts'
13
+ tokenizer = transformers.AutoTokenizer.from_pretrained(model)
14
+ model = transformers.BertForSequenceClassification.from_pretrained(model)
15
+
16
+ sentence_pairs = [("Nire ama etxera etorri da.", "Ama visitan etorri da etxera")]
17
+ for sp in sentence_pairs:
18
+ result = model(**tokenizer(*sp, return_tensors='pt'))
19
+ print(sp, result.logits)
20
+
21
+ ```