Spaces:
Sleeping
Sleeping
Commit
·
bf297c3
1
Parent(s):
6087cd7
Fix permission main.py
Browse files
main.py
CHANGED
|
@@ -17,13 +17,9 @@ def classify(request: EmailRequest):
|
|
| 17 |
# Classify email
|
| 18 |
category = classify_email(model, masked_email)
|
| 19 |
|
| 20 |
-
# Demask back to original
|
| 21 |
-
demasked_email = demask_pii(masked_email, entities)
|
| 22 |
-
|
| 23 |
return {
|
| 24 |
"input_email_body": request.input_email_body,
|
| 25 |
"list_of_masked_entities": entities,
|
| 26 |
"masked_email": masked_email,
|
| 27 |
-
"category_of_the_email": category
|
| 28 |
-
"demasked_email": demasked_email
|
| 29 |
}
|
|
|
|
| 17 |
# Classify email
|
| 18 |
category = classify_email(model, masked_email)
|
| 19 |
|
|
|
|
|
|
|
|
|
|
| 20 |
return {
|
| 21 |
"input_email_body": request.input_email_body,
|
| 22 |
"list_of_masked_entities": entities,
|
| 23 |
"masked_email": masked_email,
|
| 24 |
+
"category_of_the_email": category
|
|
|
|
| 25 |
}
|