Update app.py
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ print(prediction)
|
|
| 37 |
|
| 38 |
from sentence_transformers import CrossEncoder
|
| 39 |
model1 = CrossEncoder('cross-encoder/nli-deberta-v3-xsmall')
|
| 40 |
-
scores1 =
|
| 41 |
|
| 42 |
#Convert scores to labels
|
| 43 |
label_mapping = ['contradiction', 'entailment', 'neutral']
|
|
|
|
| 37 |
|
| 38 |
from sentence_transformers import CrossEncoder
|
| 39 |
model1 = CrossEncoder('cross-encoder/nli-deberta-v3-xsmall')
|
| 40 |
+
scores1 = model1.predict([(sentence1, sentence2)])
|
| 41 |
|
| 42 |
#Convert scores to labels
|
| 43 |
label_mapping = ['contradiction', 'entailment', 'neutral']
|