Update app.py

app.py

@@ -125,12 +125,13 @@ def preprocess_image(image_array, session_id):
 
         _model_cache['face_detector'] = face_detector
         _model_cache['face_parser'] = face_parser
+        _model_cache['facer_module'] = facer.hwc2bchw
             
     subprocess.run([
             "python", "scripts/run_preprocessing.py", "--video_or_images_path", saved_image_path
         ], check=True, capture_output=True, text=True)
     
-    segment(f'{session_id}', _model_cache['face_detector'], _model_cache['face_parser'])
+    segment(f'{session_id}', _model_cache['face_detector'], _model_cache['face_parser'], _model_cache['facer_module'])
 
     crop_dir = os.path.join(os.environ["PIXEL3DMM_PREPROCESSED_DATA"], session_id, "cropped")
     image = first_image_from_dir(crop_dir)