fix repo path

app.py

@@ -12,30 +12,34 @@ from deoldify import device
 from deoldify.device_id import DeviceId
 from deoldify.visualize import *
 # from huggingface_hub import HfApi, snapshot_download
-from huggingface_hub import HfApi
+from huggingface_hub import HfApi, HfFolder
 
 os.system("pip freeze")
 from collections.abc import Sized # Import Sized from collections.abc
 # Suppress warnings
 warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set is empty.*?")
 
-# Initialize Hugging Face API
-api = HfApi()
-
-device.set(device=DeviceId.CPU)
-
 # Private repo settings
 token = os.getenv("HF_TOKEN")
 repo_id = "afondiel/image-colorizer-deoldify"
 repo_type = "space"
 
+device.set(device=DeviceId.CPU)
+
+# Initialize Hugging Face API
+api = HfApi(token=token)
+
+HfFolder.save_token(token)
+
 # Download the snapshot from the space repository
-snapshot_folder = api.snapshot_download(repo_id=repo_id, repo_type=repo_type, token=token)
+snapshot_folder = api.snapshot_download(repo_id=repo_id, repo_type=repo_type, revision="main", token=token)
 
 device.set(device=DeviceId.GPU0)
 # Load the pre-trained model
 _colorizer = get_image_colorizer(root_folder=Path(snapshot_folder), artistic=True)
 
+print(f"root_folder: {root_folder}")
+
 def colorizer_fn(input_img, render_factor):
     """
     Colorize grayscale images/photos