Update README.md

README.md

@@ -1,5 +1,8 @@
 ---
 license: mit
+widget:
+- text: "The man turned on the faucet <mask> water flows out."
+- text: "The woman received her pension <mask> she retired."
 ---
 # roberta-temporal-predictor
 A RoBERTa-base model that is fine-tuned on the [The New York Times Annotated Corpus](https://catalog.ldc.upenn.edu/LDC2008T19)
@@ -8,7 +11,14 @@ in our ROCK framework for reasoning about commonsense causality. See our [paper]
 
 # Usage
 
-For simplicity, we implement the following TempPredictor class. Example usage using the ``TempPredictor`` class:
+You can directly use this model for filling-mask tasks, as shown in the example widget.
+However, for better temporal inference, it is recommended to symmetrize the outputs as
+$$
+P(E_1 \prec E_2) = \frac{1}{2} (f(E_1,E_2) + f(E_2,E_1))
+$$
+where ``f(E_1,E_2)`` denotes the predicted probability for ``E_1`` to occur preceding ``E_2``.
+For simplicity, we implement the following TempPredictor class that incorporate this symmetrization automatically.
+Below is an example usage for the ``TempPredictor`` class:
 ```python
 from transformers import (RobertaForMaskedLM, RobertaTokenizer)
 from src.temp_predictor import TempPredictor