PraneshJs commited on
Commit
1c96b02
·
verified ·
1 Parent(s): 4a30650

increaded more tokens

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -75,7 +75,7 @@ def caption_image(image, custom_prompt=None):
75
  inputs=input_ids,
76
  attention_mask=attention_mask,
77
  images=px,
78
- max_new_tokens=128,
79
  do_sample=False, # temperature is ignored when sampling is off
80
  )
81
 
 
75
  inputs=input_ids,
76
  attention_mask=attention_mask,
77
  images=px,
78
+ max_new_tokens=400,
79
  do_sample=False, # temperature is ignored when sampling is off
80
  )
81