gokaygokay commited on
Commit
eed61fa
1 Parent(s): 5566526

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def create_captions_rich(image):
14
  input_len = model_inputs["input_ids"].shape[-1]
15
 
16
  with torch.inference_mode():
17
- generation = model.generate(**model_inputs, repetition_penalty=1.05, max_new_tokens=256, do_sample=False)
18
  generation = generation[0][input_len:]
19
  decoded = processor.decode(generation, skip_special_tokens=True)
20
 
 
14
  input_len = model_inputs["input_ids"].shape[-1]
15
 
16
  with torch.inference_mode():
17
+ generation = model.generate(**model_inputs, repetition_penalty=1.10, max_new_tokens=256, do_sample=False)
18
  generation = generation[0][input_len:]
19
  decoded = processor.decode(generation, skip_special_tokens=True)
20