Upload generate_prompt.py with huggingface_hub

generate_prompt.py

@@ -67,8 +67,8 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
         }
         idx_real = batch_idx * batch_size + idx
         messages = data[idx_real]
-        save_['content'][1]['image'] = messages['content'][1]['image']
-        save_['content'][2]['text'] = messages['content'][2]['text']
+        save_['content'][0]['image'] = messages['content'][0]['image']
+        save_['content'][1]['text'] = messages['content'][1]['text']
 
         data_list.append(messages)
         save_list.append(save_)