Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -93,18 +93,20 @@ def entity_comb(output):
|
|
93 |
output_comb.append(entity)
|
94 |
return output_comb
|
95 |
|
96 |
-
def
|
97 |
-
|
98 |
-
|
99 |
-
|
|
|
|
|
100 |
if entity['entity_group'] not in ['CARDINAL', 'EVENT']:
|
101 |
-
|
102 |
-
|
103 |
-
|
104 |
-
|
105 |
-
|
106 |
-
|
107 |
-
return
|
108 |
def create_masked_text(input_text, entities, mask_dict):
|
109 |
masked_text = input_text
|
110 |
for entity in sorted(entities, key=lambda x: x['start'], reverse=True):
|
|
|
93 |
output_comb.append(entity)
|
94 |
return output_comb
|
95 |
|
96 |
+
def create_masked_text(input_text, entities):
|
97 |
+
# Create the mask dictionary
|
98 |
+
mask_dict = create_mask_dict(entities)
|
99 |
+
|
100 |
+
masked_text = input_text
|
101 |
+
for entity in sorted(entities, key=lambda x: x['start'], reverse=True):
|
102 |
if entity['entity_group'] not in ['CARDINAL', 'EVENT']:
|
103 |
+
# Replace the entity with its entity group from the mask dictionary
|
104 |
+
masked_text = (
|
105 |
+
masked_text[:entity['start']] +
|
106 |
+
f"<{mask_dict[entity['word']]}> " + # Use angle brackets for clarity
|
107 |
+
masked_text[entity['end']:]
|
108 |
+
)
|
109 |
+
return masked_text
|
110 |
def create_masked_text(input_text, entities, mask_dict):
|
111 |
masked_text = input_text
|
112 |
for entity in sorted(entities, key=lambda x: x['start'], reverse=True):
|