Upload generate_prompt.py with huggingface_hub

generate_prompt.py

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