Update README.md

README.md

@@ -68,7 +68,7 @@ def process_ner_output(entity_mention, inputs):
             re_input.append({"re_input": new_re_input, "arg1": ent_1, "arg2": ent_2, "input": inputs})
     return re_input
     
-def post_process_re_output(re_output, re_input, ner_output):
+def post_process_re_output(re_output, text_input, ner_output):
     final_output = []
     for idx, out in enumerate(re_output):
         if out["label"] != 'O':
@@ -77,28 +77,48 @@ def post_process_re_output(re_output, re_input, ner_output):
             tmp.pop('re_input', None)
             final_output.append(tmp)
 
-    template = {"input": re_input["input"],
+    template = {"input": text_input,
                 "entity": ner_output,
                 "relation": final_output}
 
     return template
 
->>> input = "Hugging face is a French company in New york."
->>> output = ner_pip(input) # inference NER tags
-
->>> 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"]) # for each pair of entity, predict relation
->>>     re_output.append(tmp_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"])
+text_input = 'قال وزير العدل التركي بكير بوزداغ إن أنقرة تريد 12 مشتبهاً بهم من فنلندا و 21 من السويد'
+ner_output = ner_pip(text_input) # inference NER tags
+
+re_input = process_ner_output(ner_output, text_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"]) # for each pair of entity, predict relation
+    re_output.append(tmp_re_output[0])
+
+
+
+re_ner_output = post_process_re_output(re_output, text_input, ner_output) # post process NER and relation predictions
+print("Sentence: ",re_ner_output["input"])
+print('====Entity====')
+for ent in re_ner_output["entity"]:
+  print('{}--{}'.format(ent["word"], ent["entity_group"]))
+print('====Relation====')
+for rel in re_ner_output["relation"]:
+  print('{}--{}:{}'.format(rel['arg1']['word'], rel['arg2']['word'], rel['relation_type']['label']))
+
+Sentence:  قال وزير العدل التركي بكير بوزداغ إن أنقرة تريد 12 مشتبهاً بهم من فنلندا و 21 من السويد
+====Entity====
+وزير--PER
+العدل--ORG
+التركي--GPE
+بكير بوزداغ--PER
+انقرة--GPE
+مشتبها بهم--PER
+فنلندا--GPE
+21--PER
+السويد--GPE
+====Relation====
+وزير--العدل:ORG-AFF
+مشتبها بهم--فنلندا:PHYS
+21--السويد:PHYS
 ```
 
 ### BibTeX entry and citation info