Update app.py

app.py

@@ -64,31 +64,31 @@ ex_evidence = ['Recent Research Shows Human Activity Driving Earth Towards Globa
  'The estimate of total CO 2 emissions includes biotic carbon emissions, mainly from deforestation.',
  'The vast majority of anthropogenic carbon dioxide emissions come from combustion of fossil fuels, principally coal, oil, and natural gas, with additional contributions coming from deforestation, changes in land use, soil erosion and agriculture (including livestock).']
 
-ex_labels = ['NOT_ENOUGH_INFO',
- 'SUPPORTS',
- 'NOT_ENOUGH_INFO',
- 'SUPPORTS',
- 'NOT_ENOUGH_INFO',
- 'NOT_ENOUGH_INFO',
- 'NOT_ENOUGH_INFO',
- 'NOT_ENOUGH_INFO',
- 'REFUTES',
- 'REFUTES',
- 'NOT_ENOUGH_INFO',
- 'SUPPORTS',
- 'NOT_ENOUGH_INFO',
- 'SUPPORTS',
- 'SUPPORTS',
- 'NOT_ENOUGH_INFO',
- 'NOT_ENOUGH_INFO',
- 'NOT_ENOUGH_INFO',
- 'NOT_ENOUGH_INFO',
- 'NOT_ENOUGH_INFO',
- 'REFUTES',
- 'NOT_ENOUGH_INFO',
- 'REFUTES',
- 'NOT_ENOUGH_INFO',
- 'NOT_ENOUGH_INFO']
+ex_labels = ['not enough info',
+ 'supports',
+ 'not enough info',
+ 'supports',
+ 'not enough info',
+ 'not enough info',
+ 'not enough info',
+ 'not enough info',
+ 'refutes',
+ 'refutes',
+ 'not enough info',
+ 'supports',
+ 'not enough info',
+ 'supports',
+ 'supports',
+ 'not enough info',
+ 'not enough info',
+ 'not enough info',
+ 'not enough info',
+ 'not enough info',
+ 'refutes',
+ 'not enough info',
+ 'refutes',
+ 'not enough info',
+ 'not enough info']
 
 ex_df = pd.DataFrame({'claim' : ex_claims, 'evidence' : ex_evidence, 'label': ex_labels})
 
@@ -208,25 +208,10 @@ with torch.no_grad():
    st.write("**Claim:**", option_claim)
    st.write("**Evidence**", option_evidence)
 
-   # clean up pred label
-   pred_label = "not enough info"
-   if labels[0] == 'entailment':
-   	pred_label = 'supports'
-   elif labels[0] == 'contradiction':
-   	pred_label = 'refutes'
-
    st.write("**The predicted label is**:", labels[0])
 
    # clean up true label
    true_label = list(filtered_df[filtered_df['evidence'] == option_evidence]['label'])[0]
-   true_out = true_label
-   if true_label == "NOT_ENOUGH_INFO":
-   	true_out = "not enough info"
-   elif true_label == "SUPPORTS":
-   	true_out = "supports"
-   else:
-   	true_out == "refutes"
-
    st.write("**The true label is**", true_out)
 
 st.write("Check out my github repository to try out custom claim and evidence pairs, linked under references.")