Update app.py

app.py

@@ -59,8 +59,10 @@ def predict(text: str):
         return "⚠️ Please enter a reaction description.", None, None
 
     # ── 1. Severity prediction ──
-    results   = clf_pipeline(text)[0]           # list of {label, score}
-    scores    = {r["label"]: r["score"] for r in results}
+    raw = clf_pipeline(text)
+    # Handle both [[{...}]] and [{...}] shapes
+    results = raw[0] if isinstance(raw[0], list) else raw
+    scores = {r["label"]: r["score"] for r in results}
 
     # Normalise to expected label keys
     # The model may emit "LABEL_0"/"LABEL_1" or "Non-Severe"/"Severe"