gxy commited on
Commit
df08d3f
·
1 Parent(s): d129116

Update app.py

Browse files

longer length for zh

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,7 +21,7 @@ def inference(raw_image, model_n, strategy):
21
  config = GenerationConfig(
22
  do_sample=False,
23
  num_beams=3,
24
- max_length=20,
25
  min_length=5,
26
  )
27
  captions = model.generate(**input, generation_config=config)
@@ -29,7 +29,7 @@ def inference(raw_image, model_n, strategy):
29
  config = GenerationConfig(
30
  do_sample=True,
31
  top_p=0.9,
32
- max_length=20,
33
  min_length=5,
34
  )
35
  captions = model.generate(**input, generation_config=config)
 
21
  config = GenerationConfig(
22
  do_sample=False,
23
  num_beams=3,
24
+ max_length=50,
25
  min_length=5,
26
  )
27
  captions = model.generate(**input, generation_config=config)
 
29
  config = GenerationConfig(
30
  do_sample=True,
31
  top_p=0.9,
32
+ max_length=50,
33
  min_length=5,
34
  )
35
  captions = model.generate(**input, generation_config=config)