Update gen_splits.py

gen_splits.py

@@ -4,7 +4,11 @@ import os
 import download_and_extract_zip
 
 def gen_splits():
-    download_and_extract_zip.download_and_extract_zip()
+
+    URL = os.getenv('URL')
+    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]
     loaders = []