azaninello commited on
Commit
4039b68
·
1 Parent(s): 3172561

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -9,11 +9,12 @@ tokenizer = GPT2Tokenizer.from_pretrained(
9
  )
10
 
11
  generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('gpt2'),
12
- tokenizer='gpt2',
13
  #do_sample=True,
14
  #max_length=120,
15
  #top_k=50,
16
- #top_p=0.95,)
 
17
 
18
  def generator(start_your_text = ''):
19
  shroom_result = generator(start_your_text)
 
9
  )
10
 
11
  generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('gpt2'),
12
+ tokenizer='gpt2'#,
13
  #do_sample=True,
14
  #max_length=120,
15
  #top_k=50,
16
+ #top_p=0.95,
17
+ )
18
 
19
  def generator(start_your_text = ''):
20
  shroom_result = generator(start_your_text)