Guspfc commited on
Commit
f61ded6
·
verified ·
1 Parent(s): 3e7b25e

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -16,11 +16,7 @@ def generate_caption(pixel_values):
16
  with torch.no_grad():
17
  generated_ids = model.generate(
18
  pixel_values=pixel_values,
19
- max_length=50,
20
- num_beams=4,
21
- early_stopping=True,
22
- no_repeat_ngram_size=2
23
- )
24
  return processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
25
 
26
  def translate_to_portuguese(text):
 
16
  with torch.no_grad():
17
  generated_ids = model.generate(
18
  pixel_values=pixel_values,
19
+ max_length=500)
 
 
 
 
20
  return processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
21
 
22
  def translate_to_portuguese(text):