| |
|
| | 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
|
| | }
|
| | |
