Update app.py

app.py

@@ -215,6 +215,7 @@ with st.container():
                     verdict_start_time = time.time()
                     verdict = "NEI"
                     details = ""
+                    prob2class, pred_bc = 0, "Not used"
                     prob3class, pred_tc = classify_claim(
                         claim, evidence, model_tc, tokenizer_tc,
                         "cuda" if torch.cuda.is_available() else "cpu"
@@ -226,13 +227,13 @@ with st.container():
                         )
                         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>
-                                <p><strong>3-Class Predicted Label:</strong> {['NEI', 'SUPPORTED', 'REFUTED'][pred_tc]}</p>
-                                <p><strong>2-Class Probability:</strong> {prob2class.item():.2f}</p>
-                                <p><strong>2-Class Predicted Label:</strong> {['SUPPORTED', 'REFUTED'][pred_bc]}</p>
-                            """
+                    if show_details:
+                        details = f"""
+                            <p><strong>3-Class Probability:</strong> {prob3class.item():.2f}</p>
+                            <p><strong>3-Class Predicted Label:</strong> {['NEI', 'SUPPORTED', 'REFUTED'][pred_tc]}</p>
+                            <p><strong>2-Class Probability:</strong> {prob2class.item():.2f}</p>
+                            <p><strong>2-Class Predicted Label:</strong> {['SUPPORTED', 'REFUTED'][pred_bc]}</p>
+                        """
                     verdict_time = time.time() - verdict_start_time
 
                     # Store verification history and the latest result