Update app.py

app.py

@@ -13,8 +13,8 @@ from pytorch_grad_cam.utils.image import show_cam_on_image
 
 
 WORKSPACE_ROOT = Path(
-    os.environ.get("WORKSPACE_ROOT", Path(__file__).resolve().parents[1])
-)
+    os.environ.get("WORKSPACE_ROOT", Path(__file__).resolve().parent)
+).resolve()
 SRC_DIR = WORKSPACE_ROOT / "src"
 sys.path.insert(0, str(SRC_DIR))
 
@@ -33,7 +33,7 @@ CAPTIONING_STATE = None
 
 def load_params():
     """params.yaml을 읽어서 데모, 모델, 체크포인트 설정을 가져온다."""
-    with open("params.yaml", "r", encoding="utf-8") as f:
+    with open(WORKSPACE_ROOT / "params.yaml", "r", encoding="utf-8") as f:
         return yaml.safe_load(f)
 
 # params.yaml의 demo.class_names에서 학습 당시 클래스 목록을 가져온다.