Zakia commited on
Commit
de3f434
1 Parent(s): bd92217

Updated the examples sections of the model card

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