update example

README.md

@@ -63,8 +63,8 @@ del state_dict['model.embeddings.position_ids']
 model.load_state_dict(state_dict)
 
 # use the model
-target = ['My cat died yesterday.']
-observer = ['I am sorry for your loss.']
+target = ["I'm so sad that my cat died yesterday."]
+observer = ["It's ok to feel sad."]
 
 target_encodings = tokenizer(target, padding=True, truncation=True)
 target_input_ids = torch.LongTensor(target_encodings['input_ids']).to('cuda')
@@ -75,5 +75,5 @@ observer_attention_mask = torch.LongTensor(observer_encodings['attention_mask'])
 
 model.eval()
 output = model(target_input_ids, target_attention_mask, observer_input_ids, observer_attention_mask)
-print(output) # [0.3304]
+print(output) # [0.5755]
 ```