afondiel commited on
Commit
0d9b367
1 Parent(s): 60a326d
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -19,7 +19,9 @@ warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set
19
 
20
 
21
  # private repo
22
- token = "colorizer"
 
 
23
  snapshot_folder = snapshot_download(repo_id="afondiel/image-colorizer-deoldify", token=token)
24
 
25
  # Set the device to use for computation
 
19
 
20
 
21
  # private repo
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