Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -85,7 +85,7 @@ def caption_image(image, custom_prompt=None):
85
  inputs=input_ids,
86
  attention_mask=attention_mask,
87
  images=px,
88
- max_new_tokens=128,
89
  do_sample=False, # Deterministic generation
90
  temperature=1.0,
91
  )
 
85
  inputs=input_ids,
86
  attention_mask=attention_mask,
87
  images=px,
88
+ max_new_tokens=1000,
89
  do_sample=False, # Deterministic generation
90
  temperature=1.0,
91
  )