Spanicin commited on
Commit
c639632
·
verified ·
1 Parent(s): 18c0bae

Update app_parallel.py

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