fixing model path

app.py

@@ -1,18 +1,23 @@
 import os
 import warnings
+from pathlib import Path
 
 import gradio as gr
 from deoldify import device
 from deoldify.device_id import DeviceId
 from deoldify.visualize import get_image_colorizer
+from huggingface_hub import snapshot_download
 
 os.system("pip freeze")
 warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set is empty.*?")
 
 device.set(device=DeviceId.CPU)
 
+REPO_ID = "leonelhs/deoldify"
+snapshot_folder = snapshot_download(repo_id=REPO_ID)
+
 device.set(device=DeviceId.GPU0)
-colorizer = get_image_colorizer(artistic=True)
+colorizer = get_image_colorizer(root_folder=Path(snapshot_folder), artistic=True)
 
 
 def predict(image):

models/.gitkeep

@@ -1 +0,0 @@
-

models/ColorizeArtistic_gen.pth

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