EchoTruth / modules /validation.py
lightmate's picture
added modules file
ada3cf7 verified
raw
history blame
269 Bytes
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]