Nikhil Singh commited on
Commit
832bd14
·
1 Parent(s): 1afbc3a
Files changed (1) hide show
  1. app.py +4 -1
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