llava
1import os2import json3import argparse4
5parser = argparse.ArgumentParser()6parser.add_argument("--src", type=str)7parser.add_argument("--dst", type=str)8args = parser.parse_args()9
10all_answers = []11for line_idx, line in enumerate(open(args.src)):12res = json.loads(line)13question_id = res['question_id']14text = res['text'].rstrip('.').lower()15all_answers.append({"questionId": question_id, "prediction": text})16
17with open(args.dst, 'w') as f:18json.dump(all_answers, f)19