Updated the examples sections of the model card

README.md

@@ -16,10 +16,16 @@ tags:
 - depression
 - text classification
 widget:
-- text: "This medication has changed my life for the better. I've experienced no side effects and my symptoms of depression have significantly decreased."
+- text: "After starting this new treatment, I felt an immediate improvement in my mood and energy levels."
   example_title: "Example 1"
-- text: "I've had a terrible experience with this medication. It made me feel nauseous and I didn't notice any improvement in my condition."
+- text: "I was apprehensive about the side effects of the medication, but thankfully I haven't experienced any."
   example_title: "Example 2"
+- text: "This medication has changed my life for the better. I've experienced no side effects and my symptoms of depression have significantly decreased."
+  example_title: "Example 3"
+- text: "I've had a terrible experience with this medication. It made me feel nauseous and I didn't notice any improvement in my condition."
+  example_title: "Example 4"
+- text: "Since I began taking L-methylfolate, my experience has been overwhelmingly positive with noticeable improvements."
+  example_title: "Example 5"
 ---
 
 # Model Card for Zakia/distilbert-drugscom_depression_reviews
@@ -75,15 +81,19 @@ def print_predictions(review_text, model, tokenizer):
     predictions = F.softmax(outputs.logits, dim=-1)
     # LABEL_0 is for low quality and LABEL_1 for high quality
     print(f"Review: \"{review_text}\"")
-    print(f"Prediction: {{'LABEL_0 (Low quality)': {predictions[0][0].item():.4f}, 'LABEL_1 (High quality)': {predictions[0][1].item():.4f}}}\n")
-
-# High quality review example
-high_quality_review = "This medication has changed my life for the better. I've experienced no side effects and my symptoms of depression have significantly decreased."
-print_predictions(high_quality_review, model, tokenizer)
-
-# Low quality review example
-low_quality_review = "I've had a terrible experience with this medication. It made me feel nauseous and I didn't notice any improvement in my condition."
-print_predictions(low_quality_review, model, tokenizer)
+    print(f"Prediction: {{'LABEL_0 (Low Quality)': {predictions[0][0].item():.4f}, 'LABEL_1 (High Quality)': {predictions[0][1].item():.4f}}}\n")
+
+# Example usage for various scenarios
+example_reviews = [
+    "After starting this new treatment, I felt an immediate improvement in my mood and energy levels.",
+    "I was apprehensive about the side effects of the medication, but thankfully I haven't experienced any.",
+    "This medication has changed my life for the better. I've experienced no side effects and my symptoms of depression have significantly decreased.",
+    "I've had a terrible experience with this medication. It made me feel nauseous and I didn't notice any improvement in my condition.",
+    "Since I began taking L-methylfolate, my experience has been overwhelmingly positive with noticeable improvements."
+]
+
+for review in example_reviews:
+    print_predictions(review, model, tokenizer)
 ```
 
 ## Training Details