Hugging Face's logo

Spaces:
hf-dongpyo
/
gpt2
Sleeping

App Files Community
hf-dongpyo commited on
Commit
6654daa
·
1 Parent(s): 548b1d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ 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)
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, 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):