model output

app.py

@@ -203,7 +203,7 @@ with torch.no_grad():
    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("is ", labels[0])
    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)