Update README.md
Browse files
README.md
CHANGED
@@ -28,10 +28,10 @@ pip install -U --no-cache-dir transformers
|
|
28 |
```
|
29 |
from transformers import pipeline
|
30 |
|
31 |
-
model_name: str = '
|
32 |
generator: pipeline = pipeline('text-generation', model=model_name)
|
33 |
|
34 |
-
prompt: str = '
|
35 |
gen_text: str = generator(prompt, do_sample=True, min_length=50)
|
36 |
|
37 |
print(f'Result: {gen_text}')
|
|
|
28 |
```
|
29 |
from transformers import pipeline
|
30 |
|
31 |
+
model_name: str = 'FrostAura/gpt-neo-1.3B-fiction-novel-generation'
|
32 |
generator: pipeline = pipeline('text-generation', model=model_name)
|
33 |
|
34 |
+
prompt: str = 'So far my day has been '
|
35 |
gen_text: str = generator(prompt, do_sample=True, min_length=50)
|
36 |
|
37 |
print(f'Result: {gen_text}')
|