divi7007 commited on
Commit
122ca63
1 Parent(s): 3c50c94

Update app.py

Browse files

right now the output is coming in square brackets so

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def generate(prompt, seq_len=128, temperature=0.8, filter_thres=0.9):
39
  use_tqdm=True,
40
  )
41
 
42
- decoded_output = tokenizer.batch_decode(output_tensor, skip_special_tokens=True)
43
 
44
  return decoded_output
45
 
 
39
  use_tqdm=True,
40
  )
41
 
42
+ decoded_output = tokenizer.batch_decode(output_tensor[0], skip_special_tokens=True)
43
 
44
  return decoded_output
45