model output

app.py

@@ -202,11 +202,30 @@ 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[0])
-   st.write("The claim: ", option_claim, "is ", labels, " by ", option_evidence)
+
+   st.write("**The claim**:", option_claim)
+
+   # clean up pred label
+   pred_label = "not enough info"
+   if labels[0] == 'entailment':
+   	pred_label = 'supports'
+   elif labels[0] == 'contradiction':
+   	pred_label = 'refutes'
+
+   # write out
+   st.write("**is**", labels[0])
+   st.write("**with**", option_evidence)
+
+   # clean up true label
    true_label = list(filtered_df[filtered_df['evidence'] == option_evidence]['label'])[0]
-   st.write("The correct relationship is", true_label)
+   if true_label == "NOT_ENOUGH_INFO":
+   	true_label = "not enough info"
+   elif true_label == "SUPPORTS":
+   	true_label = "supports"
+   else:
+   	true_label == "refutes"
+   	
+   st.write("**The correct relationship is**", true_label)
 
 # section 6: analysis
 st.markdown("## Critical Analysis")