afondiel commited on
Commit
ddd3994
1 Parent(s): ede4c6e

update model path

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -17,8 +17,11 @@ os.system("pip freeze")
17
  from collections.abc import Sized # Import Sized from collections.abc
18
  warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set is empty.*?")
19
 
 
 
20
  # private repo
21
- token="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"
 
22
  snapshot_folder = snapshot_download(repo_id="afondiel/image-colorizer-deoldify", token=token)
23
 
24
  # Set the device to use for computation
 
17
  from collections.abc import Sized # Import Sized from collections.abc
18
  warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set is empty.*?")
19
 
20
+
21
+
22
  # private repo
23
+ # from config import HUGGINGFACE_TOKEN as token
24
+ token = os.getenv('HUGGINGFACE_TOKEN')
25
  snapshot_folder = snapshot_download(repo_id="afondiel/image-colorizer-deoldify", token=token)
26
 
27
  # Set the device to use for computation