add label

app.py

@@ -31,7 +31,12 @@ def predict_sentiment(text):
     scores = outputs.logits[0].detach().numpy()
     predictions = torch.nn.functional.softmax(torch.tensor(scores), dim=0)
     sentiment = torch.argmax(predictions).item()
-    return sentiment, predictions[sentiment].item()
+    
+    # Map sentiment to label
+    labels = ["positive", "negative", "neutral"]
+    sentiment_label = labels[sentiment]
+    
+    return sentiment_label, predictions[sentiment].item()
 
 # Endpoint untuk analisis sentimen
 @app.post("/predict")