Updated the examples sections of the model card
Browse files
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 |
-
|
79 |
-
|
80 |
-
|
81 |
-
|
82 |
-
|
83 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
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
|