Update README.md

README.md

@@ -38,8 +38,8 @@ prompt = "What's your name?"
 
 context = "Hinata was soft-spoken and polite, always addressing people with proper honorifics. She is kind, always thinking of others more than for herself, caring for their feelings and well-being. She doesn't like being confrontational for any reason. This led to her being meek or timid to others, as her overwhelming kindness can render her unable to respond or act for fear of offending somebody."
 
-input_ids = tokenizer(f"personality: {contexts}", f"inquiry: {prompt}", return_tensors='pt').input_ids
-outputs = trained_model.generate(input_ids, num_beams=6, diversity_penalty=5.0, num_beam_groups=2)
+input_ids = tokenizer(f"personality: {context}", f"inquiry: {prompt}", return_tensors='pt').input_ids
+outputs = trained_model.generate(input_ids, num_beams=6, diversity_penalty=2.5, num_beam_groups=2)
 
 print("Answer:\t", tokenizer.decode(outputs[0], skip_special_tokens=True))
 
@@ -62,7 +62,7 @@ Which reflect its bio which was filled by GPT-3
 > "He stands primarily to gain self-esteem, which he often receives through the submission of others"
 
 
-In gist, Dr. Greenbaum tried to tease Sebastian about his seductive traits but this model's go-to response was to shut her down since the biography of Sebastian states he often try to assert his dominance over others.
+In gist, Dr. Greenbaum tried to tease Sebastian about his seductive traits but this model's go-to response was to shut her down since the biography of Sebastian states he often tries to assert his dominance over others.
 
 ```py
 prompt = dataset['test'][66]['request']