Spaces:
Sleeping
Sleeping
Nikhil Singh
commited on
Commit
·
832bd14
1
Parent(s):
1afbc3a
T5 fixx
Browse files
app.py
CHANGED
@@ -80,12 +80,15 @@ def present(email_file, labels, multilingual=False):
|
|
80 |
# Format entities for DataFrame: Convert list of dicts to list of lists
|
81 |
entities_data = [[entity['text'], entity['label']] for entity in entities]
|
82 |
|
|
|
|
|
83 |
email_info = {
|
84 |
"Subject": email.subject,
|
85 |
"From": email.from_,
|
86 |
"To": email.to,
|
87 |
"Date": email.date,
|
88 |
-
"Extracted Entities": entities_data # Adjusted for DataFrame
|
|
|
89 |
}
|
90 |
return [email_info[key] for key in ["Subject", "From", "To", "Date"]] + [entities_data]
|
91 |
|
|
|
80 |
# Format entities for DataFrame: Convert list of dicts to list of lists
|
81 |
entities_data = [[entity['text'], entity['label']] for entity in entities]
|
82 |
|
83 |
+
refined_entities = refine_entities_with_t5(entities)
|
84 |
+
|
85 |
email_info = {
|
86 |
"Subject": email.subject,
|
87 |
"From": email.from_,
|
88 |
"To": email.to,
|
89 |
"Date": email.date,
|
90 |
+
"Extracted Entities": entities_data, # Adjusted for DataFrame
|
91 |
+
"Refined Entities": refined_entities
|
92 |
}
|
93 |
return [email_info[key] for key in ["Subject", "From", "To", "Date"]] + [entities_data]
|
94 |
|