Upload infer_.py with huggingface_hub
Browse files
infer_.py
CHANGED
|
@@ -67,7 +67,7 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
|
|
| 67 |
{"type": "text", "text": "Describe this video."},
|
| 68 |
],
|
| 69 |
"answer":"",
|
| 70 |
-
"result":""
|
| 71 |
}
|
| 72 |
messages = {
|
| 73 |
"role": "user",
|
|
@@ -121,9 +121,6 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
|
|
| 121 |
generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
|
| 122 |
)
|
| 123 |
for idx,x in enumerate(output_text):
|
| 124 |
-
if x == sd_ans[idx]:
|
| 125 |
-
correct_num = correct_num + 1
|
| 126 |
-
print(x,sd_ans[idx])
|
| 127 |
data_list[idx]['result'] = x
|
| 128 |
save_data.append(data_list[idx])
|
| 129 |
|
|
|
|
| 67 |
{"type": "text", "text": "Describe this video."},
|
| 68 |
],
|
| 69 |
"answer":"",
|
| 70 |
+
"result":"",
|
| 71 |
}
|
| 72 |
messages = {
|
| 73 |
"role": "user",
|
|
|
|
| 121 |
generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
|
| 122 |
)
|
| 123 |
for idx,x in enumerate(output_text):
|
|
|
|
|
|
|
|
|
|
| 124 |
data_list[idx]['result'] = x
|
| 125 |
save_data.append(data_list[idx])
|
| 126 |
|