Add ColorizeArtistic_gen.pth model

app.py

@@ -24,22 +24,24 @@ token = os.getenv("HF_TOKEN")
 repo_id = "afondiel/image-colorizer-deoldify"
 repo_type = "space"
 
+# running primarly on CPU
 device.set(device=DeviceId.CPU)
 
 # Initialize Hugging Face API
 api = HfApi(token=token)
 
+# Authenticate
 HfFolder.save_token(token)
 
 # Download the snapshot from the space repository
-snapshot_folder = api.snapshot_download(repo_id=repo_id, repo_type=repo_type, revision="main", token=token)
+snapshot_folder = api.snapshot_download(repo_id=repo_id, repo_type=repo_type)
 
+# if GPU available
 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
@@ -73,4 +75,4 @@ demo = gr.Interface(
 
 # Launch the demo
 if __name__ == "__main__":
-    demo.launch()
+  demo.launch()

models/ColorizeArtistic_gen.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3f750246fa220529323b85a8905f9b49c0e5d427099185334d048fb5b5e22477
+size 255144681