hf-dongpyo commited on
Commit
55d9dd6
·
1 Parent(s): 6654daa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -6,7 +6,12 @@ gpt2_tkn = GPT2Tokenizer.from_pretrained('gpt2')
6
 
7
  def generate(starting_text):
8
  tkn_ids = gpt2_tkn.encode(starting_text, return_tensors = 'pt')
9
- gpt2_tensors = mdl.generate(tkn_ids, max_length = 100, no_repeat_ngram_size = True, num_beams = 3, do_sample = True)
 
 
 
 
 
10
  response = ""
11
  # response = gpt2_tensors
12
  for i, x in enumerate(gpt2_tensors):
 
6
 
7
  def generate(starting_text):
8
  tkn_ids = gpt2_tkn.encode(starting_text, return_tensors = 'pt')
9
+ gpt2_tensors = mdl.generate(tkn_ids,
10
+ max_length = 100,
11
+ no_repeat_ngram_size = True,
12
+ num_beams = 3,
13
+ do_sample = True,
14
+ temperature = 1.5)
15
  response = ""
16
  # response = gpt2_tensors
17
  for i, x in enumerate(gpt2_tensors):