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!")
|