Upload infer_1.py with huggingface_hub

infer_1.py

@@ -42,7 +42,7 @@ data = read_json('/home/zbz5349/WorkSpace/aigeeks/Qwen2.5-VL/LLaMA-Factory/data/
 save_data = []
 correct_num = 0
 begin = 0
-end = len(data)
+end = 1
 batch_size = 1
 for batch_idx in tqdm(range(begin, end, batch_size)):
     batch = data[batch_idx:batch_idx + batch_size] 
@@ -50,8 +50,8 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
     image_list = []
     input_text_list = []
     data_list = []
-    save_data = []
-    
+    save_list = []
+    sd_ans = []
     # while True:
     for idx, i in enumerate(batch):
         save_ =  {
@@ -66,7 +66,8 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
             {"type": "image", "image": "file:///path/to/image2.jpg"},
             {"type": "text", "text": "Describe this video."},
         ],
-        "answer":""
+        "answer":"None",
+        "result":"None",
     }
         messages =  {
         "role": "user",
@@ -95,8 +96,9 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
         save_['content'][1]['image'] = image_path
         save_['content'][2]['text'] = question
         save_['answer'] = answer
+        sd_ans.append(answer)
         data_list.append(messages)
-        save_data.append(save_)
+        save_list.append(save_)
 
     text = processor.apply_chat_template(data_list, tokenize=False, add_generation_prompt=True)
     image_inputs, video_inputs, video_kwargs = process_vision_info(data_list, return_video_kwargs=True)
@@ -119,11 +121,11 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
     output_text = processor.batch_decode(
         generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
     )
-    save_["answer"] = output_text
-    if output_text == answer:
-        correct_num = correct_num + 1
-    save_data.append(save_)
+    for idx,x in enumerate(output_text):
+        save_list[idx]['result'] = x
+        save_data.append(save_list[idx])
+        
 print("correct_num", correct_num)
-write_json("infer_answer.json",save_data)
+write_json("infer_answer_ori.json",save_data)