Update download_repo.py

download_repo.py

@@ -4,7 +4,7 @@ import io
 import json 
 
 
-def download_gitlab_repo_to_hfspace(api_url, project_id, version):
+def download_gitlab_repo_to_hfspace(api_url, project_id, version, target_folder):
     try:
         # Construct the URL for the release's zip file
         encoded_project_id = urllib.parse.quote(project_id, safe="")
@@ -14,7 +14,7 @@ def download_gitlab_repo_to_hfspace(api_url, project_id, version):
         response = requests.get(url, stream=True)
 
         if response.status_code == 200:
-            _upload_file_to_hfspace(response, api, DATA_DIR, HF_SPACE_NAME)
+            _upload_file_to_hfspace(response, api, target_folder, HF_SPACE_NAME)
         else:
             print(f"Failed to download the release: {response.status_code} - {response.reason}")
             print(response.text)
@@ -27,7 +27,7 @@ def download_gitlab_repo_to_hfspace(api_url, project_id, version):
         print(f"An error occurred: {e}")
 
 
-def _upload_file_to_hfspace(response, api, DATA_DIR, HF_SPACE_NAME):
+def _upload_file_to_hfspace(response, api, target_folder, HF_SPACE_NAME):
     archive_bytes = io.BytesIO(response.content)
     # Extract filename from content-disposition header
     content_disposition = response.headers.get("content-disposition")
@@ -36,7 +36,7 @@ def _upload_file_to_hfspace(response, api, DATA_DIR, HF_SPACE_NAME):
 
         api.upload_file(
             path_or_fileobj=archive_bytes,
-            path_in_repo=f"{DATA_DIR}/{filename}",
+            path_in_repo=f"{target_folder}/{filename}",
             repo_id=HF_SPACE_NAME,
             repo_type="space"
         )