updated commit 005

model/alarm_classifier.py

@@ -5,6 +5,6 @@ classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnl
 
 
 def classify_alarm(alarm_text):
-    labels = ["Power Failure", "Signal Loss", "Hardware Fault", "Configuration Issue", "Reboot"]
+    labels = ["power failure", "link down", "overheating", "hardware issue", "configuration error"]
     result = classifier(alarm_text, candidate_labels=labels)
-    return result[0]['label']
+    return result["labels"][0], result["scores"][0]