Is it possible to give the result in the first person and not in the third?
#6
by
Lisstrange
- opened
Hi! I want to ask a question:
Model return summary in third preson only.
Example: My name is Nikita
Returns: Nikita's name is Nikita...
Could I change it?
Script for summary:
tokenizer = AutoTokenizer.from_pretrained("philschmid/bart-large-cnn-samsum")
model = AutoModelForSeq2SeqLM.from_pretrained("philschmid/bart-large-cnn-samsum")
def get_summary(tokenizer, model, text):
tokens = tokenizer(text, truncation=True,
padding="longest", return_tensors="pt")
summary = model.generate(**tokens)
summary = tokenizer.decode(summary[0])
return summary
get_summary(tokenizer=tokenizer, model=model, text="My name is Nikita" )
Thanks!
Try using Flan T5 Model. They may help you with the first person summarization.