Update generic.py

generic.py

@@ -122,7 +122,8 @@ def save_image_vis(image_file_path， question):
     text_scores = (text_scores - text_scores.min()) / (text_scores.max() - text_scores.min())
     vis_data_records = [visualization.VisualizationDataRecord(text_scores, 0, 0, 0, 0, 0, model_lrp.question_tokens, 1)]
     html1 = visualization.visualize_text(vis_data_records)
-    return img, html1.data
+    answer = vqa_answers[model_lrp.output.question_answering_score.argmax()]
+    return img, html1.data, answer