BigSalmon commited on
Commit
6623bf1
·
1 Parent(s): d18493e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def extend(input_text, num_return_sequences, max_size=20, top_k=50, top_p=0.95):
60
  text = tokenizer.decode(generated_sequence, clean_up_tokenization_spaces=True)
61
  print(text)
62
  total_sequence = (
63
- input_text + text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :]
64
  )
65
  print(total_sequence)
66
 
 
60
  text = tokenizer.decode(generated_sequence, clean_up_tokenization_spaces=True)
61
  print(text)
62
  total_sequence = (
63
+ text[len(tokenizer.decode(encoded_prompt[0], clean_up_tokenization_spaces=True)) :]
64
  )
65
  print(total_sequence)
66