Use:
from transformers import pipeline
pipe = pipeline("text-classification", model="PaDaS-Lab/privacy-policy-relation-extraction", return_all_scores=True)
example = "We store your basic account information, including your name, username, and email address until you ask us to delete them."
results = pipe(example)
threshold = 0.5
print([result for result in results[0] if result['score'] >= threshold])
Performance:
- Downloads last month
- 281