kwyong commited on
Commit
bdf31eb
·
verified ·
1 Parent(s): 41473d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def predict(img):
23
  outputs = model.generate(
24
  pixel_values.to(device),
25
  min_length=1,
26
- max_new_tokens=30,
27
  bad_words_ids=[[processor.tokenizer.unk_token_id]],
28
  )
29
 
 
23
  outputs = model.generate(
24
  pixel_values.to(device),
25
  min_length=1,
26
+ max_new_tokens=20000,
27
  bad_words_ids=[[processor.tokenizer.unk_token_id]],
28
  )
29