app/models/nlp_model.py

from transformers import pipeline
from app.core.config import settings

# Initialize the NLP model pipeline
pipe = pipeline("text-classification", model=settings.MODEL_NAME)

def classify_text(text: str) -> str:
    result = pipe(text)[0]
    return result["label"]