Update app.py

app.py

@@ -200,66 +200,67 @@ with st.container():
                 # Placeholder for displaying result/loading
                 with st.spinner("Verifying..."): 
                     start_time = time.time() 
-
-                    with torch.no_grad():
-                        # Extract evidence
-                        evidence_start_time = time.time()
-                        evidence = extract_evidence_tfidf_qatc(
-                            claim, context, model_qatc, tokenizer_qatc,
-                            "cuda" if torch.cuda.is_available() else "cpu",
-                            confidence_threshold=tfidf_threshold,
-                            length_ratio_threshold=length_ratio_threshold
-                        )
-                        evidence_time = time.time() - evidence_start_time
-
-                        # Classify the claim
-                        verdict_start_time = time.time()
-                        verdict = "NEI"
-                        details = ""
-                        prob3class, pred_tc = classify_claim(
-                            claim, evidence, model_tc, tokenizer_tc,
+ 
+                    # Extract evidence
+                    evidence_start_time = time.time()
+                    evidence = extract_evidence_tfidf_qatc(
+                        claim, context, model_qatc, tokenizer_qatc,
+                        "cuda" if torch.cuda.is_available() else "cpu",
+                        confidence_threshold=tfidf_threshold,
+                        length_ratio_threshold=length_ratio_threshold
+                    )
+                    evidence_time = time.time() - evidence_start_time
+
+                    # Classify the claim
+                    verdict_start_time = time.time()
+                    verdict = "NEI"
+                    details = ""
+                    prob3class, pred_tc = classify_claim(
+                        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:
+                        prob2class, pred_bc = classify_claim(
+                            claim, evidence, model_bc, tokenizer_bc,
                             "cuda" if torch.cuda.is_available() else "cpu"
                         )
-                        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"
-                            elif prob2class > prob3class:
-                                verdict = "REFUTED"
-                            else:
-                                verdict = ["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>
-                                """
-                        verdict_time = time.time() - verdict_start_time
-
-                        # Store verification history and the latest result
-                        st.session_state.history.append({
-                            "claim": claim,
-                            "evidence": evidence,
-                            "verdict": verdict,
-                            "evidence_time": evidence_time,
-                            "verdict_time": verdict_time,
-                            "details": details
-                        })
-                        st.session_state.latest_result = {
-                            "claim": claim,
-                            "evidence": evidence,
-                            "verdict": verdict,
-                            "evidence_time": evidence_time,
-                            "verdict_time": verdict_time,
-                            "details": details
-                        }
-
-                        if torch.cuda.is_available():
-                            torch.cuda.empty_cache()
+                        if pred_bc == 0:
+                            verdict = "SUPPORTED"
+                        else:
+                            verdict = "REFUTED"
+                        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
+                    st.session_state.history.append({
+                        "claim": claim,
+                        "evidence": evidence,
+                        "verdict": verdict,
+                        "evidence_time": evidence_time,
+                        "verdict_time": verdict_time,
+                        "details": details
+                    })
+                    st.session_state.latest_result = {
+                        "claim": claim,
+                        "evidence": evidence,
+                        "verdict": verdict,
+                        "evidence_time": evidence_time,
+                        "verdict_time": verdict_time,
+                        "details": details
+                    }
+
+                    if torch.cuda.is_available():
+                        torch.cuda.empty_cache()
 
                     # Display the result after verification
                     res = st.session_state.latest_result