adrian7305 commited on
Commit
bf297c3
·
1 Parent(s): 6087cd7

Fix permission main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -5
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
  }