mwauranjorogekelvin commited on
Commit
9a88615
·
verified ·
1 Parent(s): 603d4e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -94,9 +94,9 @@ def solve(req: SolveRequest, _: str = Depends(verify_key)):
94
  pixel_values = processor(image, return_tensors="pt").pixel_values.to(DEVICE)
95
  generated_ids = model.generate(
96
  pixel_values,
97
- max_new_tokens=32,
98
- num_beams=4,
99
- early_stopping=True
100
  )
101
  text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
102
 
 
94
  pixel_values = processor(image, return_tensors="pt").pixel_values.to(DEVICE)
95
  generated_ids = model.generate(
96
  pixel_values,
97
+ max_new_tokens=10,
98
+ num_beams=1,
99
+ do_sample=False
100
  )
101
  text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
102