mwauranjorogekelvin commited on
Commit
1fa5d79
·
verified ·
1 Parent(s): 323e667

Update app.py

Browse files

tokenization update

Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -96,7 +96,8 @@ def solve(req: SolveRequest, _: str = Depends(verify_key)):
96
 
97
  generated_ids = model.generate(
98
  pixel_values,
99
- max_new_tokens=10,
 
100
  num_beams=4,
101
  do_sample=False
102
  )
 
96
 
97
  generated_ids = model.generate(
98
  pixel_values,
99
+ max_new_tokens=5,
100
+ min_new_tokens=5,
101
  num_beams=4,
102
  do_sample=False
103
  )