Spaces:
Runtime error
Runtime error
Commit
·
2ec3b65
1
Parent(s):
e8e0cce
Update app.py
Browse files
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()
|
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!")
|