Update app.py

app.py

@@ -42,9 +42,9 @@ def generate_answer_git(processor, model, image, question):
     generated_ids = model.generate(pixel_values=pixel_values, input_ids=input_ids, max_length=50, return_dict_in_generate=True, output_scores=True)
     print('scores:')
     print(generated_ids.scores)
-    scoresList0 = torch.softmax(logits[0], dim=1).flatten().tolist()
+    scoresList0 = torch.softmax(generated_ids.scores[0], dim=1).flatten().tolist()
     print(scoresList0)
-    scoresList1 = torch.softmax(logits[1], dim=1).flatten().tolist()
+    scoresList1 = torch.softmax(generated_ids.scores[1], dim=1).flatten().tolist()
     print(scoresList1)
     print('sequences:')
     print(generated_ids.sequences)