Update submission_task1.py
Browse files
medvqa/submission_samples/gi-2025/submission_task1.py
CHANGED
|
@@ -16,7 +16,7 @@ rouge = load("rouge")
|
|
| 16 |
meteor = load("meteor")
|
| 17 |
|
| 18 |
|
| 19 |
-
val_dataset = load_dataset("SimulaMet
|
| 20 |
predictions = [] # List to store predictions
|
| 21 |
|
| 22 |
gpu_name = torch.cuda.get_device_name(
|
|
|
|
| 16 |
meteor = load("meteor")
|
| 17 |
|
| 18 |
|
| 19 |
+
val_dataset = load_dataset("SimulaMet/Kvasir-VQA-test", split="validation")
|
| 20 |
predictions = [] # List to store predictions
|
| 21 |
|
| 22 |
gpu_name = torch.cuda.get_device_name(
|