abhaskumarsinha commited on
Commit
03bbf47
·
1 Parent(s): 817ea5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def pad_or_slice_tensor(tensor):
60
  return padded_or_sliced_tensor
61
 
62
  def generate_text(input_text):
63
- input_text += " [UNK]"
64
  tokens = bpe.process_line(input_text)
65
  tokens = vectorizer(tokens)[:10]
66
  tokens = pad_or_slice_tensor(tokens)
 
60
  return padded_or_sliced_tensor
61
 
62
  def generate_text(input_text):
63
+ input_text += " ]"
64
  tokens = bpe.process_line(input_text)
65
  tokens = vectorizer(tokens)[:10]
66
  tokens = pad_or_slice_tensor(tokens)