afondiel commited on
Commit
cf8ef53
1 Parent(s): e861489

update root repo path

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -22,7 +22,11 @@ warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set
22
  from dotenv import load_dotenv
23
  load_dotenv()
24
  token = os.getenv("HF_TOKEN")
25
- snapshot_folder = snapshot_download(repo_id="afondiel/image-colorizer-deoldify", token=token)
 
 
 
 
26
 
27
  # Set the device to use for computation
28
  # choices: CPU, GPU0...GPU7
 
22
  from dotenv import load_dotenv
23
  load_dotenv()
24
  token = os.getenv("HF_TOKEN")
25
+ repo_id = "afondiel/image-colorizer-deoldify"
26
+ repo_type = "space" # Specify that the repo type is 'space'
27
+
28
+ # Download the snapshot from the space repository
29
+ snapshot_folder = snapshot_download(repo_id=repo_id, repo_type=repo_type, token=token)
30
 
31
  # Set the device to use for computation
32
  # choices: CPU, GPU0...GPU7