Spaces:
Sleeping
Sleeping
Nikhil Singh
commited on
Commit
·
0f44836
1
Parent(s):
832bd14
final t5 fix
Browse files
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[
|
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 |
|