fix the model path

app.py

@@ -1,3 +1,7 @@
+import os
+import warnings
+from pathlib import Path
+
 import gradio as gr
 from gradio_imageslider import ImageSlider
 import numpy as np
@@ -7,11 +11,15 @@ import fastai
 from deoldify import device
 from deoldify.device_id import DeviceId
 from deoldify.visualize import *
+from huggingface_hub import snapshot_download
 
-import warnings
+os.system("pip freeze")
 from collections.abc import Sized # Import Sized from collections.abc
 warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set is empty.*?")
 
+
+snapshot_folder = snapshot_download(repo_id="afondiel/image-colorizer-deoldify")
+
 # Set the device to use for computation
 # choices:  CPU, GPU0...GPU7
 if torch.cuda.is_available():
@@ -20,7 +28,7 @@ else:
     device.set(device=DeviceId.CPU)
 
 # Load the pre-trained model
-colorizer = get_image_colorizer(artistic=True)
+colorizer = get_image_colorizer(root_folder=Path(snapshot_folder), artistic=True)
 
 
 def colorizer_fn(input_img, render_factor):