model output

app.py

@@ -202,7 +202,9 @@ with torch.no_grad():
    scores = model(**features).logits
    label_mapping = ['contradiction', 'entailment', 'neutral']
    labels = [label_mapping[score_max] for score_max in scores.argmax(dim=1)]
-   print("The claim: ", option_claim, "is ", labels, " by ", option_evidence)
+   st.write("The claim: ", option_claim, "is ", labels, " by ", option_evidence)
+   true_label = list(filtered_df[filtered_df['evidence'] == option_evidence]['label'])[0]
+   st.write("The correct relationship is", true_label)
 
 # section 6: analysis
 st.markdown("## Critical Analysis")