pathlib path

utils/shared_utils.py

@@ -17,7 +17,7 @@ from utils.photo_wct import PhotoWCT
 from utils.photo_smooth import Propagator
 
 # Load models
-root = pyrootutils.setup_root(Path.cwd(), pythonpath=True)
+root = Path.cwd()
 device = "cuda" if torch.cuda.is_available() else "cpu"
 # Load model
 p_wct = PhotoWCT()