updated responses to actual labels

app.py

@@ -8,6 +8,7 @@ checkpoint="MoritzLaurer/DeBERTa-v3-base-mnli-fever-anli"
 
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
 model=AutoModelForSequenceClassification.from_pretrained(checkpoint)
+answers=['Yes',"Doesn't matter","No"]
 
 title = "The Seagull story"
 description = """
@@ -68,7 +69,7 @@ def generate_tone(index,question):
     input = tokenizer(passages[index], question, truncation=True, return_tensors="pt")
     output = model(input["input_ids"].to("cpu"))  # device = "cuda:0" or "cpu"
     prediction = torch.softmax(output["logits"][0], -1).tolist()
-    return prediction
+    return answers[np.argmax(prediction)]
 
 passages=["General","Pier","Boat","Island"]