Update README.md

README.md

@@ -80,18 +80,18 @@ def post_process_re_output(re_output, re_input, ner_output):
     return template
 
 >>> input = "Hugging face is a French company in New york."
->>> output = ner_pip(input)
+>>> output = ner_pip(input) # inference NER tags
 
->>> re_input = process_ner_output(output, input)
+>>> re_input = process_ner_output(output, input) # prepare a pair of entity and predict relation type
 
 >>> re_output = []
 >>> for idx in range(len(re_input)):
->>>     tmp_re_output = re_pip(re_input[idx]["re_input"])
+>>>     tmp_re_output = re_pip(re_input[idx]["re_input"]) # for each pair of entity, predict relation
 >>>     re_output.append(tmp_re_output)
 
 
 
->>> re_ner_output = post_process_re_output(re_output)
+>>> re_ner_output = post_process_re_output(re_output) # post process NER and relation predictions
 >>> print("Sentence: ",re_ner_output["input"])
 >>> print("Entity: ", re_ner_output["entity"])
 >>> print("Relation: ", re_ner_output["relation"])