hf-dongpyo commited on
Commit
8020d1d
·
1 Parent(s): 7aaa0d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -7,7 +7,10 @@ gpt2_tkn = GPT2Tokenizer.from_pretrained('gpt2')
7
  def generate(starting_text):
8
  tkn_ids = gpt2_tkn.encode(starting_text, return_tensors = 'pt')
9
  gpt2_tensors = mdl.generate(tkn_ids)
10
- response = gpt2_tensors
 
 
 
11
 
12
  return response
13
 
 
7
  def generate(starting_text):
8
  tkn_ids = gpt2_tkn.encode(starting_text, return_tensors = 'pt')
9
  gpt2_tensors = mdl.generate(tkn_ids)
10
+ response = ""
11
+ # response = gpt2_tensors
12
+ for i, x in enumerate(gpt2_tensors):
13
+ response += f"{i}: {gpt2_tkn.decode(x, skip_special_tokens = True)}"
14
 
15
  return response
16