OsamaBsher/AITA-Reddit-Dataset
Viewer • Updated • 271k • 313 • 5
How to use OsamaBsher/AITA-LLM-Text-Generation with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("OsamaBsher/AITA-LLM-Text-Generation")
model = AutoModelForSeq2SeqLM.from_pretrained("OsamaBsher/AITA-LLM-Text-Generation")The model generates comments for moral dilemmas that are shaped into an AITA reddit post, giving a verdict and a reason for the verdict.
The model is a finetuned version of facebook/bart-base on the AITA subreddit dataset with comments
Repository for more information: https://github.com/OsamaBsher/AITA-LLM-Text-Generation
For more details see paper: https://arxiv.org/abs/2310.18336
@OsamaBsher and Ameer Sabri
@OsamaBsher