pranee31/emailClassification

1from fastapi import FastAPI2from pydantic import BaseModel3from utils import mask_pii4import joblib5 6model = joblib.load("saved_model/email_classifier.pkl")7app = FastAPI()8 9class EmailRequest(BaseModel):10    email_body: str11 12@app.post("/")13def classify_email(request: EmailRequest):14    original_email = request.email_body15    masked_email, masked_entities = mask_pii(original_email)16    category = model.predict([masked_email])[0]17 18    return {19        "input_email_body": original_email,20        "list_of_masked_entities": masked_entities,21        "masked_email": masked_email,22        "category_of_the_email": category23    }