model output

app.py

@@ -202,6 +202,8 @@ 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)]
+   st.write("**The claim:", option_claim)
+   st.write("is ", labels.values())
    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)