Aekanun commited on
Commit
bf50dd7
1 Parent(s): 9c57d07

revise app.py with 512 new tokens

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -102,7 +102,7 @@ Only return the transcription in Thai language."""
102
  with torch.no_grad():
103
  outputs = model.generate(
104
  **inputs,
105
- max_new_tokens=256,
106
  do_sample=False,
107
  pad_token_id=processor.tokenizer.pad_token_id
108
  )
 
102
  with torch.no_grad():
103
  outputs = model.generate(
104
  **inputs,
105
+ max_new_tokens=512, ##256
106
  do_sample=False,
107
  pad_token_id=processor.tokenizer.pad_token_id
108
  )