alexnasa commited on
Commit
d12cad9
·
1 Parent(s): 1dbf56f

temp folder for output

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -8,12 +8,15 @@ import shutil
8
  import time
9
  import gradio as gr
10
  import sys
 
11
 
 
 
12
 
13
  # Set environment variables
14
  os.environ["PIXEL3DMM_CODE_BASE"] = f"{os.getcwd()}"
15
- os.environ["PIXEL3DMM_PREPROCESSED_DATA"] = f"{os.getcwd()}/proprocess_results"
16
- os.environ["PIXEL3DMM_TRACKING_OUTPUT"] = f"{os.getcwd()}/tracking_results"
17
 
18
  def sh(cmd): subprocess.check_call(cmd, shell=True)
19
 
 
8
  import time
9
  import gradio as gr
10
  import sys
11
+ import tempfile
12
 
13
+ # make one unique temp folder for this run
14
+ BASE_TMP = tempfile.mkdtemp(prefix="pixel3dmm_")
15
 
16
  # Set environment variables
17
  os.environ["PIXEL3DMM_CODE_BASE"] = f"{os.getcwd()}"
18
+ os.environ["PIXEL3DMM_PREPROCESSED_DATA"] = os.path.join(BASE_TMP, "preprocess_results")
19
+ os.environ["PIXEL3DMM_TRACKING_OUTPUT"] = os.path.join(BASE_TMP, "tracking_results")
20
 
21
  def sh(cmd): subprocess.check_call(cmd, shell=True)
22