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']
|