MinhDS commited on
Commit
52f888c
·
verified ·
1 Parent(s): 9e00578

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -61,7 +61,8 @@ def run_example(task_prompt, image, text_input=None, model_id='microsoft/Florenc
61
  max_new_tokens=1024,
62
  early_stopping=False,
63
  do_sample=False,
64
- num_beams=3,
 
65
  )
66
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
67
  parsed_answer = processor.post_process_generation(
 
61
  max_new_tokens=1024,
62
  early_stopping=False,
63
  do_sample=False,
64
+ num_beams=1,
65
+ use_cache=False,
66
  )
67
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
68
  parsed_answer = processor.post_process_generation(