Update app.py

app.py

@@ -10,7 +10,7 @@ def load_classifier():
 
 classifier = load_classifier()
 
-st.title("Email Classifier")
+st.title("Email Classifier with PII Masking")
 
 # Input email
 email_input = st.text_area("Paste your email here:")
@@ -23,9 +23,10 @@ if st.button("Analyze"):
         pii_result = mask_pii(email_input)
 
         # Step 2: Predict category
-        masked_text = pii_result["masked_email"]
+        masked_text = pii_result["English_masked"]
         prediction = classifier.predict(masked_text)
         pii_result["category_of_the_email"] = prediction
+        del pii_result["English_masked"]
         # Step 3: Format full output
         # output = {
         #     "input_email_body": email_input,