1inkusFace commited on
Commit
2069f2c
·
verified ·
1 Parent(s): 2a603bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -328,9 +328,9 @@ def captioning(img, prompt_1,prompt_2,prompt_3,prompt_4,prompt_5):
328
  **inputs,
329
  do_sample=True,
330
  num_beams=5,
331
- max_length=64,
332
- #min_length=16,
333
- top_p=0.1,
334
  repetition_penalty=1.0,
335
  length_penalty=2.0,
336
  temperature=0.55,
 
328
  **inputs,
329
  do_sample=True,
330
  num_beams=5,
331
+ max_length=256,
332
+ min_length=42,
333
+ top_p=0.2,
334
  repetition_penalty=1.0,
335
  length_penalty=2.0,
336
  temperature=0.55,