Spanicin commited on
Commit
0fcce1d
·
verified ·
1 Parent(s): e562e2d

Update app_parallel.py

Browse files
Files changed (1) hide show
  1. app_parallel.py +2 -0
app_parallel.py CHANGED
@@ -221,6 +221,8 @@ def generate_audio(voice_cloning, voice_gender, text_prompt):
221
  # Preprocessing step that runs only once
222
  def run_preprocessing(args):
223
  global path_of_lm_croper, path_of_net_recon_model, dir_of_BFM_fitting
 
 
224
  fixed_temp_dir = "/tmp/preprocess_data"
225
  os.makedirs(fixed_temp_dir, exist_ok=True)
226
  preprocessed_data_path = os.path.join(fixed_temp_dir, "preprocessed_data.pkl")
 
221
  # Preprocessing step that runs only once
222
  def run_preprocessing(args):
223
  global path_of_lm_croper, path_of_net_recon_model, dir_of_BFM_fitting
224
+ first_frame_dir = os.path.join(args.result_dir, 'first_frame_dir')
225
+ os.makedirs(first_frame_dir, exist_ok=True)
226
  fixed_temp_dir = "/tmp/preprocess_data"
227
  os.makedirs(fixed_temp_dir, exist_ok=True)
228
  preprocessed_data_path = os.path.join(fixed_temp_dir, "preprocessed_data.pkl")