Update

README.md

@@ -1,5 +1,5 @@
 ---
-title: Turkish Named Entity Recognition
+title: Romanian Named Entity Recognition
 emoji: 🏃
 colorFrom: indigo
 colorTo: indigo

app.py

@@ -62,29 +62,56 @@ Run_Button = st.button("Run", key=None)
 if Run_Button == True:
     
     ner = setModel(named_persons_only = False)
-    output = ner(input_text)[0]
-
-    df = pd.DataFrame.from_dict(output)
+    output = ner(input_text)[0] # only one sentence
 
+    # tabular form
+    data = []
+    for word in output["words"]:
+        if word["tag"]!="O":
+            data.append({
+                "word": word["text"],
+                "tag": word["tag"],
+                "start_char": word["start_char"],
+                "end_char": word["end_char"],
+                "span_after": word["span_after"],
+            })
+    df = pd.DataFrame.from_dict(data)
     st.subheader("Recognized Entities")
     st.dataframe(df)
 
-    """
+
     st.subheader("Spacy Style Display")
     spacy_display = {}
     spacy_display["ents"] = []
-    spacy_display["text"] = input_text
+    spacy_display["text"] = output["text"]
     spacy_display["title"] = None
 
-    for entity in output:
-        if aggregation != "none":
-            spacy_display["ents"].append({"start": entity["start"], "end": entity["end"], "label": entity["entity_group"]})
-        else:
-            spacy_display["ents"].append({"start": entity["start"], "end": entity["end"], "label": entity["entity"]})
+    for word in output["words"]:
+        #spacy_display["ents"].append({"start": entity["start"], "end": entity["end"], "label": entity["entity_group"]})
+        spacy_display["ents"].append({"start": word["start_char"], "end": word["end_char"], "label": word["tag"]})
     
-    entity_list = ["PER", "LOC", "ORG", "MISC"]
-    colors = {'PER': '#85DCDF', 'LOC': '#DF85DC', 'ORG': '#DCDF85', 'MISC': '#85ABDF',}
+    entity_list = ['O', 'PERSON', 'ORG', 'GPE', 'LOC', 'NAT_REL_POL',
+     'EVENT', 'LANGUAGE', 'WORK_OF_ART', 'DATETIME',
+     'PERIOD', 'MONEY', 'QUANTITY', 'NUMERIC',
+     'ORDINAL', 'FACILITY']
+    colors = {
+        'O': '#FFF',
+        'PERSON': '#F00',
+        'ORG': '#F00',
+        'GPE': '#F00',
+        'LOC': '#F00',
+        'NAT_REL_POL': '#F00',
+        'EVENT': '#F00',
+        'LANGUAGE': '#F00',
+        'WORK_OF_ART': '#F00',
+        'DATETIME': '#F00',
+        'PERIOD': '#F00',
+        'MONEY': '#F00',
+        'QUANTITY': '#F00',
+        'NUMERIC': '#F00',
+        'ORDINAL': '#F00',
+        'FACILITY': '#F00',
+    }
     html = spacy.displacy.render(spacy_display, style="ent", minify=True, manual=True, options={"ents": entity_list, "colors": colors})
     style = "<style>mark.entity { display: inline-block }</style>"
-    st.write(f"{style}{get_html(html)}", unsafe_allow_html=True)
-    """
+    st.write(f"{style}{get_html(html)}", unsafe_allow_html=True)

gitattributes

@@ -1,27 +0,0 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin 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
-*.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