fix login
Browse files
app.py
CHANGED
@@ -19,7 +19,9 @@ warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set
|
|
19 |
|
20 |
|
21 |
# private repo
|
22 |
-
|
|
|
|
|
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
|