Kuaaangwen commited on
Commit
2ec3b65
1 Parent(s): e8e0cce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ if submit_button_compare:
59
  st.write(text_classification_model.config.id2label[0], ":", round(outputs[0][0].item()*100,2),"%")
60
  st.write(text_classification_model.config.id2label[2], ":", round(outputs[0][2].item()*100,2),"%")
61
 
62
- entailment_score = round(outputs[0][2].item()*100,2)
63
 
64
  if entailment_score >= threshold:
65
  st.subheader("The statements are very similar!")
 
59
  st.write(text_classification_model.config.id2label[0], ":", round(outputs[0][0].item()*100,2),"%")
60
  st.write(text_classification_model.config.id2label[2], ":", round(outputs[0][2].item()*100,2),"%")
61
 
62
+ entailment_score = round(outputs[0][2].item(),2)
63
 
64
  if entailment_score >= threshold:
65
  st.subheader("The statements are very similar!")