README

README.md

@@ -1,10 +1,10 @@
 ---
-title: TweetNERModel
-emoji: 🏢
-colorFrom: purple
-colorTo: red
+title: NER
+emoji: 🦀
+colorFrom: green
+colorTo: purple
 sdk: gradio
-sdk_version: 3.4.1
+sdk_version: 3.1.4
 app_file: app.py
 pinned: false
 ---

app.py

@@ -0,0 +1,55 @@
+import gradio as gr
+from tner import TransformersNER
+from spacy import displacy
+
+# model = TransformersNER("tner/roberta-large-ontonotes5")
+model = TransformersNER("tner/bertweet-large-tweetner7-all")
+
+examples = [
+    "Jacob Collier is a Grammy awarded artist from England.",
+    'Get the all-analog Classic Vinyl Edition of "Takin\' Off" Album from {@herbiehancock@} via {@bluenoterecords@} link below: {{URL}}',
+    "I’m so happy that the {@The New York Times@} sees in {@Mondaire Jones@} and {@Jamaal Bowman@} what the progressive grassroots in Westchester, Rockland and the Bronx sees ! They will both be extraordinary Congresspersons ! #cvhpower #nycd17 # nycd16",
+    "When Sebastian Thrun started working on self-driving cars at Google in 2007 , few people outside of the company took him seriously.",
+    "But Google is starting from behind. The company made a late push into hardware, and Apple’s Siri, available on iPhones, and Amazon’s Alexa software, which runs on its Echo and Dot devices, have clear leads in consumer adoption."
+]
+
+
+def predict(text):
+    output = model.predict([text])
+    tokens = output['input'][0]
+
+    def retain_char_position(p):
+        if p == 0:
+            return 0
+        return len(' '.join(tokens[:p])) + 1
+
+    doc = {
+        "text": text,
+        "ents": [{
+            "start": retain_char_position(entity['position'][0]),
+            "end": retain_char_position(entity['position'][-1]) + len(entity['entity'][-1]),
+            "label": entity['type']
+        } for entity in output['entity_prediction'][0]],
+        "title": None
+    }
+
+    html = displacy.render(doc, style="ent", page=True, manual=True, minify=True)
+    html = (
+        "<div style='max-width:100%; max-height:360px; overflow:auto'>"
+        + html
+        + "</div>"
+    )
+    
+    return html
+
+
+demo = gr.Interface(
+    fn=predict,
+    inputs=gr.inputs.Textbox(
+        lines=5,
+        placeholder="Input sentence...",
+    ),
+    outputs="html",
+    examples=examples
+)
+demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+spacy
+tner
+allennlp==2.10.0