Upload generate_prompt.py with huggingface_hub

generate_prompt.py

@@ -55,6 +55,9 @@ correct_num = 0
 begin = args.begin
 end = args.end
 batch_size = args.batch_size
+json_path = args.prompt_path
+
+
 for batch_idx in tqdm(range(begin, end, batch_size)):
     batch = data[batch_idx:min(batch_idx+batch_size, end)] 
     
@@ -109,8 +112,9 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
         idx_real = batch_idx * batch_size + idx
         save_list[idx]['result'] = x
         save_data.append(save_list[idx])
-
-json_path = args.prompt_path
+    if batch_idx % 4 ==0:
+        write_json(json_path,save_data)
+        
 write_json(json_path,save_data)