Spaces:

SimulaMet
/

MedVQA

Running

SushantGautam commited on 16 days ago

Commit

f836573

unverified ·

1 Parent(s): e521e5e

Update submission_task1.py

Files changed (1) hide show

medvqa/submission_samples/gi-2025/submission_task1.py CHANGED Viewed

@@ -16,7 +16,7 @@ rouge = load("rouge")
 meteor = load("meteor")
-val_dataset = load_dataset("SimulaMet-HOST/Kvasir-VQA")['raw'].select(range(5))
 predictions = []  # List to store predictions
 gpu_name = torch.cuda.get_device_name(

 meteor = load("meteor")
+val_dataset = load_dataset("SimulaMet/Kvasir-VQA-test", split="validation")
 predictions = []  # List to store predictions
 gpu_name = torch.cuda.get_device_name(