Update app.py

app.py

@@ -112,9 +112,9 @@ def perform_verification(claim, context, model_qatc, tokenizer_qatc, model_tc, t
         "verdict": verdict,
         "evidence_time": evidence_time,
         "verdict_time": verdict_time,
-        "prob3class": prob3class,
+        "prob3class": prob3class.item() if isinstance(prob3class, torch.Tensor) else prob3class,
         "pred_tc": pred_tc,
-        "prob2class": prob2class,
+        "prob2class": prob2class.item() if isinstance(prob2class, torch.Tensor) else prob2class,
         "pred_bc": pred_bc
     }
 
@@ -341,7 +341,7 @@ with st.container():
                             3-Class Probability: {result['prob3class']:.2f}
                             3-Class Predicted Label: {['NEI', 'SUPPORTED', 'REFUTED'][result['pred_tc']]}
                             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'}
+                            2-Class Predicted Label: {['SUPPORTED', 'REFUTED'][result['pred_bc']] if isinstance(result['pred_bc'].item(), int) and result['pred_tc'].item() != 0 else 'Not used'}
                         """
                     
                     st.session_state.latest_result = {