Updated example texts to match the GPT RLHF model examples

README.md

@@ -16,9 +16,9 @@ tags:
 - depression
 - text generation
 widget:
-- text: This medication has changed my life for the better
+- text: After starting this new treatment, I felt
   example_title: Example 1
-- text: I've had a terrible experience with this medication
+- text: I was apprehensive about the side effects of
   example_title: Example 2
 ---
 
@@ -75,11 +75,11 @@ def generate_review(prompt, model, tokenizer):
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
 # Example usage
-prompt = "This medication has changed my life for the better"
+prompt = "After starting this new treatment, I felt"
 print(generate_review(prompt, model, tokenizer))
 
 # Example usage
-prompt = "I've had a terrible experience with this medication"
+prompt = "I was apprehensive about the side effects of"
 print(generate_review(prompt, model, tokenizer))
 ```