sunbv56 commited on
Commit
a2a8d8e
·
verified ·
1 Parent(s): 22fe62c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def process_vqa(image: Image.Image, question: str):
34
 
35
  generated_ids = model.generate(
36
  **model_inputs,
37
- max_new_tokens=1024,
38
  do_sample=False,
39
  temperature=1.0,
40
  eos_token_id=processor.tokenizer.eos_token_id,
 
34
 
35
  generated_ids = model.generate(
36
  **model_inputs,
37
+ max_new_tokens=128,
38
  do_sample=False,
39
  temperature=1.0,
40
  eos_token_id=processor.tokenizer.eos_token_id,