Wendy-Fly commited on
Commit
949fea9
·
verified ·
1 Parent(s): d263f76

Upload infer_1.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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['conversations'][0]['content']
83
- answer = i['conversations'][1]['content']
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