Update README.md

README.md

@@ -18,6 +18,32 @@ The input to the model is defined as:
 [CLS] cand. question [q] gold answer [r] pred answer [c] context
 ```
 
+
+# Generation 
+
+You can use the following script to get the semantic similarity of the predicted answer given the gold answer, context, and question.
+
+```
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+sp_scorer = AutoModelForSequenceClassification.from_pretrained('alirezamsh/quip-512-mocha')
+tokenizer_sp = AutoTokenizer.from_pretrained('alirezamsh/quip-512-mocha')
+sp_scorer.eval()
+
+pred_answer = ""
+gold_answer = ""
+question = ""
+context = ""
+
+input_sp = f"{question} <q> {gold_answer} <r>" \
+                   f" {pred_answer} <c> {context}"
+
+inputs = tokenizer_sp(input_sp, max_length=512, truncation=True, \
+                                   padding="max_length", return_tensors="pt")
+
+outputs = sp_scorer(input_ids=inputs["input_ids"], attention_mask=inputs["attention_mask"])
+print(outputs)
+```
+
 # Citations
 
 ```