Update app.py

app.py

@@ -155,12 +155,12 @@ st.markdown(
     }
     .stButton>button {
         background-color: var(--primary-color);
-        color: var(--text-color); /* Đảm bảo màu chữ tương phản với nền */
+        color: red; 
         font-size: 16px;
         width: 100%;
         border-radius: 8px;
         padding: 10px;
-        border: 1px solid var(--primary-color); /* Thêm viền để nút nổi bật hơn */
+        border: 2px solid red;
     }
     .stTextArea textarea {
         font-size: 16px;
@@ -338,9 +338,9 @@ with st.container():
                     details = ""
                     if show_details:
                         details = f"""
-                            3-Class Probability: {result['prob3class'].item():.2f}
+                            3-Class Probability: {result['prob3class']:.2f}
                             3-Class Predicted Label: {['NEI', 'SUPPORTED', 'REFUTED'][result['pred_tc']]}
-                            2-Class Probability: {result['prob2class'].item():.2f}
+                            2-Class Probability: {result['prob2class']:.2f}
                             2-Class Predicted Label: {['SUPPORTED', 'REFUTED'][result['pred_bc']] if isinstance(result['pred_bc'], int) and result['pred_tc'] != 0 else 'Not used'}
                         """