Zakia commited on
Commit
1760e13
1 Parent(s): c1e6ef9

Updated the examples sections of the model card

Browse files
Files changed (1) hide show
  1. README.md +19 -7
README.md CHANGED
@@ -21,6 +21,12 @@ widget:
21
  example_title: Example 1
22
  - text: I was apprehensive about the side effects of
23
  example_title: Example 2
 
 
 
 
 
 
24
  ---
25
 
26
  # Model Card for Zakia/gpt2-drugscom_depression_reviews-hq-v1
@@ -78,13 +84,19 @@ def generate_high_quality_review(prompt, model, tokenizer):
78
  outputs = model.generate(**inputs)
79
  return tokenizer.decode(outputs[0], skip_special_tokens=True)
80
 
81
- # Example usage to generate a high-quality drug review for improved generation
82
- prompt = "After starting this new treatment, I felt"
83
- print(generate_high_quality_review(prompt, model, tokenizer))
84
-
85
- # Example usage to generate a high-quality drug review for side effect discussion
86
- prompt = "I was apprehensive about the side effects of"
87
- print(generate_high_quality_review(prompt, model, tokenizer))
 
 
 
 
 
 
88
  ```
89
 
90
  ## Training Details
 
21
  example_title: Example 1
22
  - text: I was apprehensive about the side effects of
23
  example_title: Example 2
24
+ - text: This medication has changed my life for the better
25
+ example_title: Example 3
26
+ - text: I've had a terrible experience with this medication
27
+ example_title: Example 4
28
+ - text: Since I began taking L-methylfolate, my experience has been
29
+ example_title: Example 5
30
  ---
31
 
32
  # Model Card for Zakia/gpt2-drugscom_depression_reviews-hq-v1
 
84
  outputs = model.generate(**inputs)
85
  return tokenizer.decode(outputs[0], skip_special_tokens=True)
86
 
87
+ # Example usage for various scenarios
88
+ prompts = [
89
+ "After starting this new treatment, I felt",
90
+ "I was apprehensive about the side effects of",
91
+ "This medication has changed my life for the better",
92
+ "I've had a terrible experience with this medication",
93
+ "Since I began taking L-methylfolate, my experience has been"
94
+ ]
95
+
96
+ for prompt in prompts:
97
+ print(f"Prompt: {prompt}")
98
+ print(generate_high_quality_review(prompt, model, tokenizer))
99
+ print()
100
  ```
101
 
102
  ## Training Details