Update app.py

app.py

@@ -39,7 +39,7 @@ def generate_answer_git(processor, model, image, question):
     input_ids = [processor.tokenizer.cls_token_id] + input_ids
     input_ids = torch.tensor(input_ids).unsqueeze(0)
     
-    generated_ids = model.generate(pixel_values=pixel_values, input_ids=input_ids, max_length=50, output_scores=True)
+    generated_ids = model.generate(pixel_values=pixel_values, input_ids=input_ids, max_length=50, return_dict_in_generate=True, output_scores=True)
     print(generated_ids.scores)
     generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)