|
|
| from fastapi import FastAPI
|
| from pydantic import BaseModel
|
| from models import load_model, classify_email
|
| from utils import mask_pii_multilingual
|
|
|
| app = FastAPI()
|
| tokenizer, model, device = load_model()
|
|
|
| class EmailInput(BaseModel):
|
| input_email_body: str
|
|
|
| @app.post("/classify")
|
| async def classify_route(request: EmailInput):
|
| text = request.input_email_body
|
| masked_text, entities = mask_pii_multilingual(text)
|
| category = classify_email(masked_text, tokenizer, model, device)
|
| return {
|
| "input_email_body": text,
|
| "list_of_masked_entities": entities,
|
| "masked_email": masked_text,
|
| "category_of_the_email": category
|
| }
|
| |
