FrostAura commited on
Commit
c877cfd
1 Parent(s): 0e09957

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 = 'EleutherAI/gpt-neo-1.3B'
32
  generator: pipeline = pipeline('text-generation', model=model_name)
33
 
34
- prompt: str = 'EleutherAI has'
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}')