Update README.md

README.md

@@ -10,6 +10,7 @@ pipeline_tag: text-classification
 base_model: "distilbert/distilbert-base-uncased"
 widget:
  - text: "Left context. [SEP] Right context."
+ - text: "They have 6 grandchildren. [SEP] Ane is currently coaching Crestwood High School's Boys Varsity Soccer."
 ---
 
 # DistilBERT Cross Segment Document Chunking
@@ -43,16 +44,16 @@ model = AutoModelForSequenceClassification.from_pretrained(
 
 pairs = [
     "Left context. [SEP] Right context.",
-    "he also earned five mvp stars with the martian men's tenis team in 2149. [SEP] mart jhones spent the 2166 and 2167 seasons with the all stars intergalactic in the interstelar soccer league ( isl ).",
+    "They have 6 grandchildren. [SEP] Ane is currently coaching Crestwood High School's Boys Varsity Soccer.",
 ]
 pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)
 
 pipe(pairs)
 
-[[{'label': 'SAME', 'score': 0.9845659136772156},
-  {'label': 'DIFFERENT', 'score': 0.015434039756655693}],
- [{'label': 'SAME', 'score': 0.44031277298927307},
-  {'label': 'DIFFERENT', 'score': 0.5596872568130493}]]
+[[{'label': 'SAME', 'score': 0.986},
+  {'label': 'DIFFERENT', 'score': 0.015}],
+ [{'label': 'SAME', 'score': 0.212},
+  {'label': 'DIFFERENT', 'score': 0.788}]]
 ```
 
 ## Training Data