Update download_repo_to_huggingface.py

download_repo_to_huggingface.py

@@ -4,7 +4,7 @@ import io
 import json 
 
 
-def download_and_upload_kadiAPY_repo_to_huggingfacespace(api_url, project_id, version):
+def download_gitlab_repo_to_hfspace(api_url, project_id, version):
     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_and_upload_kadiAPY_repo_to_huggingfacespace(api_url, project_id, ve
         response = requests.get(url, stream=True)
 
         if response.status_code == 200:
-            _extract_and_upload_file(response, api, DATA_DIR, HF_SPACE_NAME)
+            _upload_file_to_hfspace(response, api, DATA_DIR, HF_SPACE_NAME)
         else:
             print(f"Failed to download the release: {response.status_code} - {response.reason}")
             print(response.text)
@@ -27,18 +27,16 @@ def download_and_upload_kadiAPY_repo_to_huggingfacespace(api_url, project_id, ve
         print(f"An error occurred: {e}")
 
 
-def _extract_and_upload_file(response, api, DATA_DIR, HF_SPACE_NAME):
-    """Extracts the archive content and uploads the file."""
+def _upload_file_to_hfspace(response, api, DATA_DIR, HF_SPACE_NAME):
     archive_bytes = io.BytesIO(response.content)
     # Extract filename from content-disposition header
     content_disposition = response.headers.get("content-disposition")
     if content_disposition and "filename=" in content_disposition:
         filename = content_disposition.split("filename=")[-1].strip('"')
-        
-        # Upload the file
+
         api.upload_file(
             path_or_fileobj=archive_bytes,
             path_in_repo=f"{DATA_DIR}/{filename}",
             repo_id=HF_SPACE_NAME,
             repo_type="space"
-        )
+        )