Upload infer_1.py with huggingface_hub
Browse files- infer_1.py +2 -2
infer_1.py
CHANGED
|
@@ -79,8 +79,8 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
|
|
| 79 |
|
| 80 |
video_path = i['videos']
|
| 81 |
image_path = i['images']
|
| 82 |
-
question = i['
|
| 83 |
-
answer = i['
|
| 84 |
messages['content'][0]['video'] = video_path
|
| 85 |
messages['content'][1]['image'] = image_path
|
| 86 |
messages['content'][2]['text'] = question
|
|
|
|
| 79 |
|
| 80 |
video_path = i['videos']
|
| 81 |
image_path = i['images']
|
| 82 |
+
question = i['messages'][0]['content']
|
| 83 |
+
answer = i['messages'][1]['content']
|
| 84 |
messages['content'][0]['video'] = video_path
|
| 85 |
messages['content'][1]['image'] = image_path
|
| 86 |
messages['content'][2]['text'] = question
|