Update app.py

app.py

@@ -104,7 +104,7 @@ st.markdown(
     }
     .stSidebar .stSlider {
         margin-bottom: 20px;
-    }
+    }   
     .stSidebar .stSelectbox {
         margin-bottom: 20px;
     }
@@ -219,19 +219,13 @@ with st.container():
                         claim, evidence, model_tc, tokenizer_tc,
                         "cuda" if torch.cuda.is_available() else "cpu"
                     )
-                    if pred_tc == 1:
-                        verdict = "SUPPORTED"
-                    elif pred_tc == 2:
-                        verdict = "REFUTED"
-                    else:
+                    if pred_tc != 0:
                         prob2class, pred_bc = classify_claim(
                             claim, evidence, model_bc, tokenizer_bc,
                             "cuda" if torch.cuda.is_available() else "cpu"
                         )
-                        if pred_bc == 0:
-                            verdict = "SUPPORTED"
-                        else:
-                            verdict = "REFUTED"
+                        verdict = "SUPPORTED" if pred_bc == 0 else "REFUTED" if prob2class > prob3class else ["NEI", "SUPPORTED", "REFUTED"][pred_tc]
+
                         if show_details:
                             details = f"""
                                 <p><strong>3-Class Probability:</strong> {prob3class.item():.2f}</p>