Update app.py

app.py

@@ -14,7 +14,8 @@ def get_sentiment():
 def get_classifier():
     global classifier
     if not classifier:
-        classifier = pipeline("text-classification",
+        classifier = pipeline(
+            "zero-shot-classification",
             model="textattack/distilbert-base-uncased-ag-news")
     return classifier
 
@@ -43,8 +44,13 @@ def process(text, features):
         sent = get_sentiment()(text)[0]
         result["sentiment"] = {"label": sent["label"], "score": sent["score"]}
     if "Classification" in features:
-        cls = get_classifier()(text)[0]
-        result["classification"] = {"label": cls["label"], "score": cls["score"]}
+        candidate_labels = [
+            "technology", "sports", "business", "politics",
+            "health", "science", "travel", "entertainment"
+        ]
+        cls = get_classifier()(text, candidate_labels=candidate_labels)
+        # Map labels → scores
+        result["classification"] = dict(zip(cls["labels"], cls["scores"]))
     if "Entities" in features:
         ents = get_ner()(text)
         result["entities"] = [