AnjaliSarawgi commited on
Commit
e6ba862
·
1 Parent(s): 864fa38
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -402,8 +402,8 @@ def predict_and_score_once(image: Image.Image, line_id: int = 1, topk: int = TOP
402
  try:
403
  out = model.generate(
404
  pixel_values,
405
- max_length=MAX_LEN,
406
- num_beams=5,
407
  do_sample=False,
408
  return_dict_in_generate=True,
409
  output_scores=True,
 
402
  try:
403
  out = model.generate(
404
  pixel_values,
405
+ max_length=100,
406
+ num_beams=1,
407
  do_sample=False,
408
  return_dict_in_generate=True,
409
  output_scores=True,