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_[0]['content'][0]['image'] = messages['content'][0]['image']
-        save_[0]['content'][1]['text'] = messages['content'][1]['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_)
@@ -110,7 +110,7 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
         generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
     )
     #print(output_text)
-    save_[0]['result'] = output_text
+    save_['result'] = output_text
     save_data.append(save_)
 
     if batch_idx % 4 ==0: