File size: 269 Bytes
ada3cf7
 
 
 
 
 
1
2
3
4
5
6
7
from transformers import pipeline

def calculate_truthfulness_score(info, context):
    classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
    result = classifier(info, [context], multi_label=False)
    return result['scores'][0]