rjiang12 commited on
Commit
3d28551
·
1 Parent(s): a33e6de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -47,8 +47,8 @@ def generate_answer_git(processor, model, image, question):
47
  print(generated_ids)
48
  generated_answer = processor.batch_decode(generated_ids.sequences, skip_special_tokens=True)
49
  print(generated_answer)
50
- generated_answer1 = processor.batch_decode(generated_ids.sequences, skip_special_tokens=True)[0]
51
- print(generated_answer1)
52
 
53
  return 'haha'
54
 
 
47
  print(generated_ids)
48
  generated_answer = processor.batch_decode(generated_ids.sequences, skip_special_tokens=True)
49
  print(generated_answer)
50
+ generated_score = processor.batch_decode(generated_ids.scores, skip_special_tokens=True)
51
+ print(generated_scores)
52
 
53
  return 'haha'
54