Commit
·
69a7750
1
Parent(s):
0300379
lets see how the hihgt works
Browse files
app.py
CHANGED
@@ -55,7 +55,8 @@ def extract_entities(sentence):
|
|
55 |
print(f"Original text: {sentence}")
|
56 |
print("Results:", results)
|
57 |
# it should look like:
|
58 |
-
# [{'entity': 'org.ent.pressagency.Reuters', 'score': np.float32(98.47),
|
|
|
59 |
for entity in results:
|
60 |
entity_span = (entity["start"], entity["end"])
|
61 |
|
@@ -66,7 +67,9 @@ def extract_entities(sentence):
|
|
66 |
entity["start"] : entity["end"]
|
67 |
].strip() # Ensure we're working with the correct portion of the text
|
68 |
entity["surface"] = entity_text
|
69 |
-
entity[
|
|
|
|
|
70 |
# print(f"Entity text: {entity}")
|
71 |
|
72 |
entities.append(entity)
|
@@ -75,6 +78,7 @@ def extract_entities(sentence):
|
|
75 |
# entities = sorted(entities, key=lambda x: x["start"])
|
76 |
return {"text": sentence, "entities": entities}
|
77 |
|
|
|
78 |
# Create Gradio interface
|
79 |
def ner_app_interface():
|
80 |
input_sentence = gr.Textbox(
|
|
|
55 |
print(f"Original text: {sentence}")
|
56 |
print("Results:", results)
|
57 |
# it should look like:
|
58 |
+
# [{'entity': 'org.ent.pressagency.Reuters', 'score': np.float32(98.47),
|
59 |
+
# 'index': 78, 'text': 'Reuters', 'start': 440, 'end': 447}]
|
60 |
for entity in results:
|
61 |
entity_span = (entity["start"], entity["end"])
|
62 |
|
|
|
67 |
entity["start"] : entity["end"]
|
68 |
].strip() # Ensure we're working with the correct portion of the text
|
69 |
entity["surface"] = entity_text
|
70 |
+
label = f"{entity['type']} ({entity.get('title', 'No title')}, {entity.get('name', 'No name')}, {entity.get('function', 'No function')})"
|
71 |
+
|
72 |
+
entity["entity"] = label
|
73 |
# print(f"Entity text: {entity}")
|
74 |
|
75 |
entities.append(entity)
|
|
|
78 |
# entities = sorted(entities, key=lambda x: x["start"])
|
79 |
return {"text": sentence, "entities": entities}
|
80 |
|
81 |
+
|
82 |
# Create Gradio interface
|
83 |
def ner_app_interface():
|
84 |
input_sentence = gr.Textbox(
|