csuvikv commited on
Commit
25aafaf
·
1 Parent(s): 5f72984
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import os
3
  from transformers import pipeline, set_seed
4
 
5
 
6
- generator = pipeline('text-generation', model='gpt2')
7
  set_seed(42)
8
 
9
 
 
3
  from transformers import pipeline, set_seed
4
 
5
 
6
+ generator = pipeline('text-generation', model='gpt2', return_full_text=False)
7
  set_seed(42)
8
 
9