npro65 commited on
Commit
f124140
·
verified ·
1 Parent(s): 7ed2f0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -35,9 +35,7 @@ def image_qa_pipeline(image, question):
35
  output_ids = model.generate(
36
  **inputs,
37
  max_new_tokens=64,
38
- num_beams=1,
39
- do_sample=True,
40
- temperature=0.7
41
  )
42
 
43
  answer = processor.tokenizer.decode(output_ids[0], skip_special_tokens=True)
 
35
  output_ids = model.generate(
36
  **inputs,
37
  max_new_tokens=64,
38
+ num_beams=3,
 
 
39
  )
40
 
41
  answer = processor.tokenizer.decode(output_ids[0], skip_special_tokens=True)