Update generic.py

generic.py

@@ -119,11 +119,14 @@ def save_image_vis(image_file_path, question):
     cv2.imwrite(
         'lxmert/experiments/paper/new.jpg', img)
     img = Image.open('lxmert/experiments/paper/new.jpg')
+    img = img.resize([448, 300])
+    orig_image = Image.open(image_file_path)
+    orig_image = orig_image.resize([448, 300])
     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)
     answer = vqa_answers[model_lrp.output.question_answering_score.argmax()]
-    return img, html1.data, answer
+    return orig_image, img, html1.data, answer