xynull commited on
Commit
cbe8be1
·
1 Parent(s): 50c6d0e

Add application file

Browse files
Files changed (1) hide show
  1. gpt2.py +3 -1
gpt2.py CHANGED
@@ -1,4 +1,6 @@
1
  from transformers import pipeline, set_seed
2
  generator = pipeline('text-generation', model='gpt2')
3
  set_seed(42)
4
- generator("Hello, I'm a language model,", max_length=30, num_return_sequences=5)
 
 
 
1
  from transformers import pipeline, set_seed
2
  generator = pipeline('text-generation', model='gpt2')
3
  set_seed(42)
4
+ r = generator("Hello, I'm a language model,", max_length=30, num_return_sequences=5)
5
+
6
+ print(r)