model/alarm_classifier.py

from transformers import pipeline

classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")


def classify_alarm(alarm_text):
    labels = ["power failure", "link down", "overheating", "hardware issue", "configuration error"]
    result = classifier(alarm_text, candidate_labels=labels)
    return result["labels"][0], result["scores"][0]