Duplicate from pie/Joint-NER-and-Relation-Extraction

Co-authored-by: Christoph Alt <[email protected]>

.gitattributes

@@ -0,0 +1,27 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,14 @@
+---
+title: Joint NER And Relation Extraction
+emoji: 🐠
+colorFrom: yellow
+colorTo: red
+sdk: gradio
+sdk_version: 2.9.4
+app_file: app.py
+pinned: false
+python_version: 3.9.12
+duplicated_from: pie/Joint-NER-and-Relation-Extraction
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference

app.py

@@ -0,0 +1,61 @@
+import re
+import gradio as gr
+from dataclasses import dataclass
+from prettytable import PrettyTable
+
+from pytorch_ie.annotations import LabeledSpan, BinaryRelation
+from pytorch_ie.auto import AutoPipeline
+from pytorch_ie.core import AnnotationList, annotation_field
+from pytorch_ie.documents import TextDocument
+
+from typing import List
+
+
+@dataclass
+class ExampleDocument(TextDocument):
+    entities: AnnotationList[LabeledSpan] = annotation_field(target="text")
+    relations: AnnotationList[BinaryRelation] = annotation_field(target="entities")
+
+
+ner_model_name_or_path = "pie/example-ner-spanclf-conll03"
+re_model_name_or_path = "pie/example-re-textclf-tacred"
+
+ner_pipeline = AutoPipeline.from_pretrained(ner_model_name_or_path, device=-1, num_workers=0)
+re_pipeline = AutoPipeline.from_pretrained(re_model_name_or_path, device=-1, num_workers=0)
+
+
+def predict(text):
+    document = ExampleDocument(text)
+
+    ner_pipeline(document)
+
+    while len(document.entities.predictions) > 0:
+        document.entities.append(document.entities.predictions.pop(0))
+
+    re_pipeline(document)
+
+    t = PrettyTable()
+    t.field_names = ["head", "tail", "relation"]
+    t.align = "l"
+    for relation in document.relations.predictions:
+        t.add_row([str(relation.head), str(relation.tail), relation.label])
+
+    html = t.get_html_string(format=True)
+    html = (
+        "<div style='max-width:100%; max-height:360px; overflow:auto'>"
+        + html
+        + "</div>"
+    )
+    
+    return html
+
+
+iface = gr.Interface(
+    fn=predict,
+    inputs=gr.inputs.Textbox(
+        lines=5,
+        default="There is still some uncertainty that Musk - also chief executive of electric car maker Tesla and rocket company SpaceX - will pull off his planned buyout.",
+    ),
+    outputs="html",
+)
+iface.launch()

requirements.txt

@@ -0,0 +1,2 @@
+pytorch-ie==0.15.0
+prettytable