SushantGautam commited on
Commit
0aa9589
·
1 Parent(s): 0563150

Add import for batched from itertools in submission_task1.py

Browse files
medvqa/submission_samples/medico-2026/submission_task1.py CHANGED
@@ -1,6 +1,7 @@
1
  from transformers import AutoModelForCausalLM
2
  from datasets import load_dataset, Image as HfImage
3
  from transformers import AutoProcessor
 
4
  import torch
5
  import json
6
  import time
 
1
  from transformers import AutoModelForCausalLM
2
  from datasets import load_dataset, Image as HfImage
3
  from transformers import AutoProcessor
4
+ from itertools import batched
5
  import torch
6
  import json
7
  import time