update model path
Browse files
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
|
|
|
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
|