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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -42,7 +42,10 @@ 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(type(generated_ids.scores))
 
 
 
46
  print('sequences:')
47
  print(generated_ids.sequences)
48
  print(generated_ids)
 
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
+ scoresList0 = torch.softmax(logits[0], dim=1).flatten().tolist()
46
+ print(scoresList0)
47
+ scoresList1 = torch.softmax(logits[1], dim=1).flatten().tolist()
48
+ print(scoresList1)
49
  print('sequences:')
50
  print(generated_ids.sequences)
51
  print(generated_ids)