Update app.py

app.py

@@ -205,7 +205,7 @@ def predict_sentiment(model, sentence, vocab, label_mapping=None):
     with torch.no_grad():
         logits = model(tensor, length).squeeze(0)
         probs = F.softmax(logits, dim=-1).cpu().tolist()
-        probs = [round(p, 3) for p in probs] 
+        probs = [round(p, 2) for p in probs] 
     idx = int(torch.tensor(probs).argmax())
     return (label_mapping[idx], probs) if label_mapping else (idx, probs)
 
@@ -266,9 +266,9 @@ def process_input(text_input, file):
     df2 = pd.DataFrame(results)
 
     styler = df2.style.format({
-        "Khả năng tiêu cực": "{:.2%}",
-        "Khả năng bình thường": "{:.2%}",
-        "Khả năng tích cực": "{:.2%}",
+        "Khả năng tiêu cực": "{:.0%}",
+        "Khả năng bình thường": "{:.0%}",
+        "Khả năng tích cực": "{:.0%}",
     })
 
     return styler