back to original

app.py

@@ -8,15 +8,12 @@ import shutil
 import time
 import gradio as gr
 import sys
-import tempfile
 
-# make one unique temp folder for this run
-BASE_TMP = tempfile.mkdtemp(prefix="pixel3dmm_")
 
 # Set environment variables
 os.environ["PIXEL3DMM_CODE_BASE"] = f"{os.getcwd()}"
-os.environ["PIXEL3DMM_PREPROCESSED_DATA"] = os.path.join(BASE_TMP, "preprocess_results")
-os.environ["PIXEL3DMM_TRACKING_OUTPUT"]  = os.path.join(BASE_TMP, "tracking_results")
+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)