Daniton commited on
Commit
90de567
·
1 Parent(s): cf909ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,8 +26,8 @@ def inference_caption(image):
26
  im,
27
  generation_type="beam_search",
28
  top_p=1.0,
29
- min_seq_len=20,
30
- seq_len=30,
31
  repetition_penalty=1.2
32
  )
33
  return open_clip.decode(generated[0].detach()).split("<end_of_text>")[0].replace("<start_of_text>", "")
 
26
  im,
27
  generation_type="beam_search",
28
  top_p=1.0,
29
+ min_seq_len=5,
30
+ seq_len=100,
31
  repetition_penalty=1.2
32
  )
33
  return open_clip.decode(generated[0].detach()).split("<end_of_text>")[0].replace("<start_of_text>", "")