fix model path
Browse files
app.py
CHANGED
@@ -17,8 +17,9 @@ 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 |
-
|
21 |
-
|
|
|
22 |
|
23 |
# Set the device to use for computation
|
24 |
# choices: CPU, GPU0...GPU7
|
|
|
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
|
25 |
# choices: CPU, GPU0...GPU7
|