Nikhil Singh commited on
Commit
0f44836
·
1 Parent(s): 832bd14

final t5 fix

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -90,7 +90,7 @@ def present(email_file, labels, multilingual=False):
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
 
95
  labels = ["PERSON", "PRODUCT", "DEAL", "ORDER", "ORDER PAYMENT METHOD", "STORE", "LEGAL ENTITY", "MERCHANT", "FINANCIAL TRANSACTION", "UNCATEGORIZED", "DATE"]
96
 
 
90
  "Extracted Entities": entities_data, # Adjusted for DataFrame
91
  "Refined Entities": refined_entities
92
  }
93
+ return [email_info["Subject"], email_info["From"], email_info["To"], email_info["Date"], entities_data, refined_entities]
94
 
95
  labels = ["PERSON", "PRODUCT", "DEAL", "ORDER", "ORDER PAYMENT METHOD", "STORE", "LEGAL ENTITY", "MERCHANT", "FINANCIAL TRANSACTION", "UNCATEGORIZED", "DATE"]
96