bart-cause-effect / README.md
taskload's picture
Update README.md
d03eced

Bart for Cause-Effect Extraction

This model was created using by Taskload, a group led by Henry Leonardi for automating information extraction. To get involved contact me at leonardi.henry@gmail.com

cause_effect = pipeline("summarization", model="taskload/bart-cause-effect")
output = cause_effect("text: "+event, max_length=300, min_length=30, do_sample=False)