Update app.py

app.py

@@ -31,6 +31,11 @@ except Exception as e:
     logger.error(f"Failed to load model: {e}")
     classifier = None
 
+LABEL_MAP = {
+    "LABEL_0": "NEGATIVE",
+    "LABEL_1": "POSITIVE"
+}
+
 class TextInput(BaseModel):
     text: str
 
@@ -79,10 +84,18 @@ def predict(input: TextInput):
     
     try:
         result = classifier(input.text)[0]
+        
+        raw_label = result["label"]
+        sentiment = LABEL_MAP.get(raw_label, raw_label)
+        
+        confidence_score = result["score"]
+        confidence_percentage = f"{confidence_score * 100:.2f}%"
+        
         return {
             "text": input.text,
-            "sentiment": result["label"],
-            "confidence": round(result["score"], 4)
+            "sentiment": sentiment,
+            "confidence": confidence_percentage,
+            "confidence_score": round(confidence_score, 4)
         }
     except Exception as e:
         logger.error(f"Prediction error: {e}")