Update app.py

app.py

@@ -202,8 +202,11 @@ features = tokenizer(option_claim, option_evidence,
 model.eval()
 with torch.no_grad():
    scores = model(**features).logits
-   label_mapping = ['contradiction', 'entailment', 'neutral']
+   label_mapping = ['refutes', 'supports', 'not enough info']
    labels = [label_mapping[score_max] for score_max in scores.argmax(dim=1)]
+   
+   st.write("**Claim:**", option_claim)
+   st.write("**Evidence**", option_evidence)
 
    # clean up pred label
    pred_label = "not enough info"
@@ -212,7 +215,7 @@ with torch.no_grad():
    elif labels[0] == 'contradiction':
    	pred_label = 'refutes'
 
-   st.write("**The predicted label is**:", pred_label)
+   st.write("**The predicted label is**:", labels[0])
 
    # clean up true label
    true_label = list(filtered_df[filtered_df['evidence'] == option_evidence]['label'])[0]