1inkusFace commited on
Commit
ea114b4
·
verified ·
1 Parent(s): 37e2aa5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -264,8 +264,8 @@ def captioning(img):
264
  **inputsa,
265
  do_sample=False,
266
  num_beams=5,
267
- max_length=256,
268
- min_length=64,
269
  top_p=0.9,
270
  repetition_penalty=1.5,
271
  length_penalty=1.0,
@@ -284,8 +284,8 @@ def captioning(img):
284
  **inputs,
285
  do_sample=False,
286
  num_beams=5,
287
- max_length=64,
288
- min_length=24,
289
  top_p=0.9,
290
  repetition_penalty=1.5,
291
  length_penalty=1.0,
@@ -304,8 +304,8 @@ def captioning(img):
304
  **inputf,
305
  do_sample=False,
306
  num_beams=5,
307
- max_length=256,
308
- min_length=24,
309
  top_p=0.9,
310
  repetition_penalty=1.5,
311
  length_penalty=1.0,
 
264
  **inputsa,
265
  do_sample=False,
266
  num_beams=5,
267
+ max_length=768,
268
+ min_length=256,
269
  top_p=0.9,
270
  repetition_penalty=1.5,
271
  length_penalty=1.0,
 
284
  **inputs,
285
  do_sample=False,
286
  num_beams=5,
287
+ max_length=96,
288
+ min_length=16,
289
  top_p=0.9,
290
  repetition_penalty=1.5,
291
  length_penalty=1.0,
 
304
  **inputf,
305
  do_sample=False,
306
  num_beams=5,
307
+ max_length=192,
308
+ min_length=96,
309
  top_p=0.9,
310
  repetition_penalty=1.5,
311
  length_penalty=1.0,