Fixes XNLI labels

app.py

@@ -54,9 +54,9 @@ def load_model_pair_classification(text, model_url_pair_classification):
     tokenizer = AutoTokenizer.from_pretrained(model_url)
     nlp = pipeline("text-classification", model=model, tokenizer=tokenizer)
     result = nlp(f"{text}</s>")
-    if result[0]["label"] == "LABEL_0":
+    if result[0]["label"] == "entailment":
         return f"Entailment: {result[0]['score']:02f}"
-    if result[0]["label"] == "LABEL_1":
+    if result[0]["label"] == "neutral":
         return f"Neutral: {result[0]['score']:02f}"
     return f"Contradiction: {result[0]['score']:02f}"