software-si commited on
Commit
4ca0e19
·
verified ·
1 Parent(s): 89d2729

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -0
README.md CHANGED
@@ -11,6 +11,9 @@ datasets:
11
  - software-si/horeca-nli
12
  pipeline_tag: text-classification
13
  library_name: sentence-transformers
 
 
 
14
  ---
15
 
16
  # CrossEncoder based on cross-encoder/nli-deberta-v3-base
@@ -50,6 +53,7 @@ Then you can load this model and run inference.
50
  ```python
51
  from sentence_transformers import CrossEncoder
52
 
 
53
  # Download from the 🤗 Hub
54
  model = CrossEncoder("software-si/kitchen-nli")
55
  # Get scores for pairs of texts
@@ -63,6 +67,9 @@ pairs = [
63
  scores = model.predict(pairs)
64
  print(scores.shape)
65
  # (5, 3)
 
 
 
66
  ```
67
 
68
  <!--
 
11
  - software-si/horeca-nli
12
  pipeline_tag: text-classification
13
  library_name: sentence-transformers
14
+ license: apache-2.0
15
+ language:
16
+ - en
17
  ---
18
 
19
  # CrossEncoder based on cross-encoder/nli-deberta-v3-base
 
53
  ```python
54
  from sentence_transformers import CrossEncoder
55
 
56
+
57
  # Download from the 🤗 Hub
58
  model = CrossEncoder("software-si/kitchen-nli")
59
  # Get scores for pairs of texts
 
67
  scores = model.predict(pairs)
68
  print(scores.shape)
69
  # (5, 3)
70
+ label_mapping = ['contradiction', 'entailment', 'neutral']
71
+
72
+
73
  ```
74
 
75
  <!--