Commit
·
878041e
1
Parent(s):
7925784
Upload ai.py
Browse files
ai.py
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from transformers import pipeline
|
| 2 |
+
|
| 3 |
+
generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
|
| 4 |
+
|
| 5 |
+
prompt = 'The thing I hate most' # can be anything
|
| 6 |
+
res = generator(prompt, max_length=50, do_sample=True, temperature=0.7)
|
| 7 |
+
|
| 8 |
+
print(res[0]['generated_text'])
|