Update gen_splits.py

gen_splits.py

@@ -9,8 +9,8 @@ def gen_splits():
     destination_folder = os.getcwd()
     download_and_extract_zip.download_and_extract_zip(URL, destination_folder)
     
-    file_paths = os.listdir('Model_TS')
-    new_file_paths = [os.getcwd() +"/Model_TS/"+ file for file in file_paths]
+    file_paths = os.listdir('Model_TS_Full')
+    new_file_paths = [os.getcwd() +"/Model_TS_Full/"+ file for file in file_paths]
     loaders = []
     for file_path in new_file_paths:
         if file_path.lower().endswith(".pdf"):