alfonsovelp commited on
Commit
0af1348
1 Parent(s): e21a4b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def generate(
55
  stream = model.generate(inputs, **generate_kwargs)
56
  output = ""
57
 
58
- decoded = tokenizer.batch_decode(generated_ids)
59
  print(decoded[0])
60
  return decoded[0]
61
 
 
55
  stream = model.generate(inputs, **generate_kwargs)
56
  output = ""
57
 
58
+ decoded = tokenizer.batch_decode(stream)
59
  print(decoded[0])
60
  return decoded[0]
61