bupa1018 commited on
Commit
ff7881c
·
1 Parent(s): 312e934

Update download_repo.py

Browse files
Files changed (1) hide show
  1. download_repo.py +3 -7
download_repo.py CHANGED
@@ -30,7 +30,7 @@ def download_gitlab_repo_to_hfspace(api_url, project_id, version, target_folder)
30
  if target_path in existing_files:
31
  print(f"File '{target_path}' already exists in the repository. Skipping upload...")
32
  else:
33
- _upload_file_to_hfspace(response, api, target_folder, HF_SPACE_NAME)
34
 
35
 
36
  except FileNotFoundError:
@@ -44,16 +44,12 @@ def download_gitlab_repo_to_hfspace(api_url, project_id, version, target_folder)
44
 
45
 
46
 
47
- def _upload_file_to_hfspace(response, api, target_folder, HF_SPACE_NAME):
48
  archive_bytes = io.BytesIO(response.content)
49
  # Extract filename from content-disposition header
50
- content_disposition = response.headers.get("content-disposition")
51
- if content_disposition and "filename=" in content_disposition:
52
- filename = content_disposition.split("filename=")[-1].strip('"')
53
-
54
  api.upload_file(
55
  path_or_fileobj=archive_bytes,
56
- path_in_repo=f"{target_folder}/{filename}",
57
  repo_id=HF_SPACE_NAME,
58
  repo_type="space"
59
  )
 
30
  if target_path in existing_files:
31
  print(f"File '{target_path}' already exists in the repository. Skipping upload...")
32
  else:
33
+ _upload_file_to_hfspace(response, api, target_path, HF_SPACE_NAME)
34
 
35
 
36
  except FileNotFoundError:
 
44
 
45
 
46
 
47
+ def _upload_file_to_hfspace(response, api, target_path, HF_SPACE_NAME):
48
  archive_bytes = io.BytesIO(response.content)
49
  # Extract filename from content-disposition header
 
 
 
 
50
  api.upload_file(
51
  path_or_fileobj=archive_bytes,
52
+ path_in_repo= target_path,
53
  repo_id=HF_SPACE_NAME,
54
  repo_type="space"
55
  )