AlexL0701 commited on
Commit
020c079
·
verified ·
1 Parent(s): b966d11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -1,19 +1,22 @@
1
- from huggingface_hub import snapshot_download, upload_folder
2
  import os
 
3
 
4
  SOURCE_REPO = "yunmorning/broken-model"
5
- TARGET_REPO = "AlexL0701/broken-model-fixed"
 
 
 
 
 
 
6
 
7
  local_dir = "/tmp/broken-model"
8
 
9
- # Download model to Space storage
10
  snapshot_download(
11
  repo_id=SOURCE_REPO,
12
- local_dir=local_dir,
13
- local_dir_use_symlinks=False
14
  )
15
 
16
- # Upload to repo
17
  upload_folder(
18
  folder_path=local_dir,
19
  repo_id=TARGET_REPO,
 
 
1
  import os
2
+ from huggingface_hub import snapshot_download, upload_folder, login
3
 
4
  SOURCE_REPO = "yunmorning/broken-model"
5
+ TARGET_REPO = "AlexL0701/broken-model-fixed" # <-- keep your repo here
6
+
7
+ # Log in using the Space secret
8
+ token = os.environ.get("HF_TOKEN")
9
+ if not token:
10
+ raise ValueError("HF_TOKEN secret not found. Add it in Space Settings > Secrets.")
11
+ login(token=token)
12
 
13
  local_dir = "/tmp/broken-model"
14
 
 
15
  snapshot_download(
16
  repo_id=SOURCE_REPO,
17
+ local_dir=local_dir
 
18
  )
19
 
 
20
  upload_folder(
21
  folder_path=local_dir,
22
  repo_id=TARGET_REPO,