Update app.py

app.py

@@ -338,9 +338,9 @@ with st.container():
                     details = ""
                     if show_details:
                         details = f"""
-                            3-Class Probability: {result['prob3class']:.2f}
+                            3-Class Probability: {result['prob3class'].item():.2f}
                             3-Class Predicted Label: {['NEI', 'SUPPORTED', 'REFUTED'][result['pred_tc']]}
-                            2-Class Probability: {result['prob2class']:.2f}
+                            2-Class Probability: {result['prob2class'].item():.2f}
                             2-Class Predicted Label: {['SUPPORTED', 'REFUTED'][result['pred_bc']] if isinstance(result['pred_bc'], int) and result['pred_tc'] != 0 else 'Not used'}
                         """