Update train_finetune.py

train_finetune.py

@@ -7,8 +7,8 @@ tokenizer = T5Tokenizer.from_pretrained("t5-base")
 
 # Load your dataset files
 data_files = {
-    "train": "dementia_train_large.json",
-    "test": "dementia_test_large.json"
+     "train": "dementia_train_combined.json",
+    "test": "dementia_test_combined.json"
 }
 dataset = load_dataset("json", data_files=data_files)