services/zero_shot.py

# services/zero_shot.py

from transformers import pipeline
import streamlit as st

@st.cache_resource
def load_zero_shot_model():
    return pipeline(
        "zero-shot-classification",
        model="facebook/bart-large-mnli"
    )

def classify_intent(text: str):
    classifier = load_zero_shot_model()

    labels = [
        "sales",
        "lead generation",
        "brand awareness",
        "engagement",
        "traffic"
    ]

    result = classifier(text, labels)

    return {
        "label": result["labels"][0],
        "score": float(result["scores"][0])
    }