llava

1
import os
2
import json
3
import argparse
4

5
parser = argparse.ArgumentParser()
6
parser.add_argument("--src", type=str)
7
parser.add_argument("--dst", type=str)
8
args = parser.parse_args()
9

10
cur_result = {}
11

12
for line in open(args.src):
13
    data = json.loads(line)
14
    qid = data['question_id']
15
    cur_result[f'v1_{qid}'] = data['text']
16

17
with open(args.dst, 'w') as f:
18
    json.dump(cur_result, f, indent=2)
19