Update app.py

app.py

@@ -11,9 +11,9 @@ import sys
 
 
 # Set environment variables
-os.environ["CODE_BASE"] = "."
-os.environ["PREPROCESSED_DATA"] = "./proprocess_results"
-os.environ["TRACKING_OUTPUT"] = "./tracking_results"
+os.environ["PIXEL3DMM_CODE_BASE"] = f"{os.getcwd()}"
+os.environ["PIXEL3DMM_PREPROCESSED_DATA"] = f"{os.getcwd()}/proprocess_results"
+os.environ["PIXEL3DMM_TRACKING_OUTPUT"] = f"{os.getcwd()}/tracking_results"
 
 def sh(cmd): subprocess.check_call(cmd, shell=True)
 
@@ -22,8 +22,6 @@ sh("pip install -e .")
 sh("cd src/pixel3dmm/preprocessing/facer && pip install -e .")
 sh("cd src/pixel3dmm/preprocessing/PIPNet/FaceBoxesV2/utils && sh make.sh")
 
-import os
-print("Current working dir:", os.getcwd())
 
 # Utility to stitch frames into a video
 def make_video_from_frames(frames_dir, out_path, fps=15):