Update src/utils.py

src/utils.py

@@ -30,9 +30,12 @@ def load_colorization_model():
     """
     # Paths to model architecture, weights, and cluster centers
     base_path = os.path.join(os.path.dirname(__file__), "models")
-    proto_file = "models/colorization_deploy_v2.prototxt"
-    model_file = "models/colorization_release_v2.caffemodel"
-    cluster_file = "models/pts_in_hull.npy"
+    # proto_file = "models/colorization_deploy_v2.prototxt"
+    # model_file = "models/colorization_release_v2.caffemodel"
+    # cluster_file = "models/pts_in_hull.npy"
+    proto_file = os.path.join(base_path, "colorization_deploy_v2.prototxt")
+    model_file = os.path.join(base_path, "colorization_release_v2.caffemodel")
+    cluster_file = os.path.join(base_path, "pts_in_hull.npy")
 
 
     # Load the model using OpenCV DNN module