oscmansan commited on
Commit
06d95bd
·
1 Parent(s): 271f0b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -22,8 +22,7 @@ def predict(image: Image.Image, question: str) -> str:
22
  generated_ids = model.generate(
23
  pixel_values=pixel_values,
24
  input_ids=input_ids,
25
- max_new_tokens=50,
26
- num_beams=5
27
  )
28
 
29
  answer = model.text_processor.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
 
22
  generated_ids = model.generate(
23
  pixel_values=pixel_values,
24
  input_ids=input_ids,
25
+ max_new_tokens=50
 
26
  )
27
 
28
  answer = model.text_processor.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()