update root repo path
Browse files
app.py
CHANGED
@@ -22,7 +22,11 @@ warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set
|
|
22 |
from dotenv import load_dotenv
|
23 |
load_dotenv()
|
24 |
token = os.getenv("HF_TOKEN")
|
25 |
-
|
|
|
|
|
|
|
|
|
26 |
|
27 |
# Set the device to use for computation
|
28 |
# choices: CPU, GPU0...GPU7
|
|
|
22 |
from dotenv import load_dotenv
|
23 |
load_dotenv()
|
24 |
token = os.getenv("HF_TOKEN")
|
25 |
+
repo_id = "afondiel/image-colorizer-deoldify"
|
26 |
+
repo_type = "space" # Specify that the repo type is 'space'
|
27 |
+
|
28 |
+
# Download the snapshot from the space repository
|
29 |
+
snapshot_folder = snapshot_download(repo_id=repo_id, repo_type=repo_type, token=token)
|
30 |
|
31 |
# Set the device to use for computation
|
32 |
# choices: CPU, GPU0...GPU7
|