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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,13 +42,13 @@ def generate_answer_git(processor, model, image, question):
42
  generated_ids = model.generate(pixel_values=pixel_values, input_ids=input_ids, max_length=50, return_dict_in_generate=True, output_scores=True)
43
  print('scores:')
44
  print(generated_ids.scores)
 
45
  print('sequences:')
46
  print(generated_ids.sequences)
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
 
 
42
  generated_ids = model.generate(pixel_values=pixel_values, input_ids=input_ids, max_length=50, return_dict_in_generate=True, output_scores=True)
43
  print('scores:')
44
  print(generated_ids.scores)
45
+ print(type(generated_ids.scores))
46
  print('sequences:')
47
  print(generated_ids.sequences)
48
  print(generated_ids)
49
  generated_answer = processor.batch_decode(generated_ids.sequences, skip_special_tokens=True)
50
  print(generated_answer)
51
+
 
52
 
53
  return 'haha'
54