Daniton commited on
Commit
1137a8a
·
1 Parent(s): 13f2790

Update app.py

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