Update generic.py

generic.py

@@ -118,10 +118,11 @@ def save_image_vis(image_file_path， question):
     # img = Image.fromarray(np.uint8(img)).convert('RGB')
     cv2.imwrite(
         'lxmert/lxmert/experiments/paper/new.jpg', img)
+    img = Image.open('lxmert/lxmert/experiments/paper/new.jpg')
     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 html1.data
+    return img, html1.data