Update app.py

app.py

@@ -1,3 +1,4 @@
+# streamlit_app.py
 import streamlit as st
 from models import load_model
 from utils import mask_pii
@@ -6,7 +7,7 @@ st.set_page_config(page_title="📧 Email Classification System", page_icon="
 
 st.title("📧 Email Classification and PII Masking System")
 
-# Load the model
+# Load model
 model = load_model("model/classifier.pkl")
 
 # Prediction function
@@ -20,7 +21,7 @@ def classify_email_direct(email_body):
         "category_of_the_email": category
     }
 
-# UI
+# Streamlit UI
 email_input = st.text_area("✉️ Enter your email text:")
 
 if st.button("🚀 Classify Email"):
@@ -36,4 +37,4 @@ if st.button("🚀 Classify Email"):
         st.subheader("📂 Predicted Category")
         st.success(result['category_of_the_email'])
     else:
-        st.warning("⚠️ Please enter some email text to classify.")
+        st.warning("⚠️ Please enter some text first.")